diff options
author | Biswakalyan Bhuyan <biswa@surgot.in> | 2024-11-27 17:12:37 +0530 |
---|---|---|
committer | Biswakalyan Bhuyan <biswa@surgot.in> | 2024-11-27 17:12:37 +0530 |
commit | 02875a1d560cf483e9362db32aae81b50665e5ec (patch) | |
tree | d288a4c6918b64bd0ad3721d8756872131ce715b /main.py | |
parent | 9f56376d64bf3307b18cd67d0fe8aaf5a6860f11 (diff) | |
download | autopredict-02875a1d560cf483e9362db32aae81b50665e5ec.tar.gz autopredict-02875a1d560cf483e9362db32aae81b50665e5ec.tar.bz2 autopredict-02875a1d560cf483e9362db32aae81b50665e5ec.zip |
Some minor changes
Diffstat (limited to 'main.py')
-rw-r--r-- | main.py | 4 |
1 files changed, 3 insertions, 1 deletions
@@ -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() |