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 19:55:12 +02:00
parent 8976e3ad75
commit 9fdca1cb31
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
IUM_06-metrics.py
View File

@ -6,9 +6,10 @@ from math import sqrt
data = pd.read_csv('beer_review_sentiment_predictions.csv') data = pd.read_csv('beer_review_sentiment_predictions.csv')
y_pred = data['Predictions'] y_pred = data['Predictions']
y_test = data['Actual'] y_test = data['Actual']
y_test_binary = (y_test >= 3).astype(int)
# Calculate metrics # Calculate metrics
accuracy = accuracy_score(y_test, y_pred.round()) accuracy = accuracy_score(y_test_binary, y_pred.round())
precision, recall, f1, _ = precision_recall_fscore_support(y_test, y_pred.round(), average='micro') precision, recall, f1, _ = precision_recall_fscore_support(y_test, y_pred.round(), average='micro')
rmse = sqrt(mean_squared_error(y_test, y_pred)) rmse = sqrt(mean_squared_error(y_test, y_pred))