From 02875a1d560cf483e9362db32aae81b50665e5ec Mon Sep 17 00:00:00 2001 From: Biswakalyan Bhuyan Date: Wed, 27 Nov 2024 17:12:37 +0530 Subject: Some minor changes --- main.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index d01d421..6fbb0c2 100644 --- a/main.py +++ b/main.py @@ -88,7 +88,7 @@ def train_model(df): # Save the model joblib.dump(best_model, 'model.pkl') - print("Model saved as 'best_mileage_predictor.pkl'") + print("Model saved as 'model.pkl'") # Main Function def main(): @@ -98,8 +98,10 @@ def main(): df = preprocess_data(df) print("Data preprocessing complete.") + print("Model is training...") train_model(df) + print("Model is trained.") if __name__ == "__main__": main() -- cgit v1.2.3-59-g8ed1b