s464937/ium_464937
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test

Browse Source
This commit is contained in:
Szymon Bartanowicz 2024-04-23 22:21:03 +02:00
parent f29fd743c3
commit 9dca2d4283
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
predict.py
View File

@ -23,6 +23,5 @@ preprocessor = ColumnTransformer(
X_test_transformed = preprocessor.fit_transform(X_test) X_test_transformed = preprocessor.fit_transform(X_test)
predictions = loaded_model.predict(X_test_transformed) predictions = loaded_model.predict(X_test_transformed)
predictions_df = pd.DataFrame(predictions, columns=['predicted_TotalKg']) predictions_df = pd.DataFrame(predictions, columns=['predicted_TotalKg'])
predictions_df.to_csv('powerlifting_test_predictions.csv', index=False) predictions_df.to_csv('powerlifting_test_predictions.csv', index=False)