14 lines
426 B
Python
14 lines
426 B
Python
import pandas as pd
|
|
from sklearn.metrics import accuracy_score, precision_recall_fscore_support, mean_squared_error
|
|
from math import sqrt
|
|
import sys
|
|
|
|
data = pd.read_csv('powerlifting_test_predictions.csv')
|
|
y_pred = data['predicted_TotalKg']
|
|
y_test = data['actual_TotalKg']
|
|
|
|
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"{build_number},{rmse}\n") |