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)