This commit is contained in:
Szymon Bartanowicz 2024-05-18 19:43:36 +02:00
parent 4a7fe811f5
commit 7c8fe37562

View File

@ -6,12 +6,9 @@ import sys
data = pd.read_csv('powerlifting_test_predictions.csv')
y_pred = data['predicted_TotalKg']
y_test = data['actual_TotalKg']
y_test_binary = (y_test >= 3).astype(int)
# build_number = sys.argv[1]
build_number = 1
build_number = sys.argv[1]
rmse = sqrt(mean_squared_error(y_test, y_pred))
with open(r"metrics.txt", "a") as f:
f.write(f"{123},{1}\n")
f.write(f"{build_number},{rmse}\n")