diff options
Diffstat (limited to 'main.py')
-rw-r--r-- | main.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -90,7 +90,7 @@ def train_model(df, target, model_name): # Save the model model_file = f'{model_name}.pkl' joblib.dump(best_model, model_file) - print("Model saved as '{model_file}'") + print(f"Model saved as '{model_file}'") # Main Function def main(): |