ium_464979/IUM_06-metrics.py

19 lines
658 B
Python
Raw Normal View History

2024-05-06 09:17:27 +02:00
import pandas as pd
2024-05-06 19:39:44 +02:00
from sklearn.metrics import accuracy_score, precision_recall_fscore_support, mean_squared_error
from math import sqrt
2024-05-06 09:17:27 +02:00
2024-05-06 19:39:44 +02:00
# Load the predictions data
data = pd.read_csv('beer_review_sentiment_predictions.csv')
y_pred = data['Predictions']
y_test = data['Actual']
2024-05-06 09:17:27 +02:00
2024-05-06 19:39:44 +02:00
# Calculate metrics
accuracy = accuracy_score(y_test, y_pred.round())
precision, recall, f1, _ = precision_recall_fscore_support(y_test, y_pred.round(), average='micro')
rmse = sqrt(mean_squared_error(y_test, y_pred))
2024-05-06 09:17:27 +02:00
2024-05-06 19:39:44 +02:00
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}')