s464979/ium_464979
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

IUM_06

Browse Source
This commit is contained in:
AWieczarek 2024-05-06 20:28:28 +02:00
parent 09d4b0772f
commit 845249c6c5
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
IUM_06-metrics.py
View File

@ -1,6 +1,7 @@
import pandas as pd
from sklearn.metrics import accuracy_score, precision_recall_fscore_support, mean_squared_error
from math import sqrt
import sys
# Load the predictions data
data = pd.read_csv('beer_review_sentiment_predictions.csv')
@ -8,6 +9,8 @@ y_pred = data['Predictions']
y_test = data['Actual']
y_test_binary = (y_test >= 3).astype(int)
build_number = sys.argv[1]
# Calculate metrics
accuracy = accuracy_score(y_test_binary, y_pred.round())
precision, recall, f1, _ = precision_recall_fscore_support(y_test_binary, y_pred.round(), average='micro')
@ -17,4 +20,7 @@ print(f'Accuracy: {accuracy}')
print(f'Micro-avg Precision: {precision}')
print(f'Micro-avg Recall: {recall}')
print(f'F1 Score: {f1}')
print(f'RMSE: {rmse}')
print(f'RMSE: {rmse}')
with open(r"beer_metrics.txt", "a") as f:
f.write(f"{accuracy},{build_number}\n")