From 9dca2d4283dcbcc76823a3f408897cfe8bf55e02 Mon Sep 17 00:00:00 2001 From: Szymon Bartanowicz Date: Tue, 23 Apr 2024 22:21:03 +0200 Subject: [PATCH] test --- predict.py | 1 - 1 file changed, 1 deletion(-) diff --git a/predict.py b/predict.py index 1273567..0cd688f 100644 --- a/predict.py +++ b/predict.py @@ -23,6 +23,5 @@ preprocessor = ColumnTransformer( X_test_transformed = preprocessor.fit_transform(X_test) predictions = loaded_model.predict(X_test_transformed) - predictions_df = pd.DataFrame(predictions, columns=['predicted_TotalKg']) predictions_df.to_csv('powerlifting_test_predictions.csv', index=False)