summaryrefslogtreecommitdiffstats
path: root/main.py
diff options
context:
space:
mode:
Diffstat (limited to 'main.py')
-rw-r--r--main.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/main.py b/main.py
index a8bdec8..b0d6673 100644
--- a/main.py
+++ b/main.py
@@ -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():