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:58:49 +02:00
parent 9fdca1cb31
commit 09d4b0772f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
IUM_06-metrics.py
View File

@ -10,7 +10,7 @@ y_test_binary = (y_test >= 3).astype(int)
# Calculate metrics
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_binary, y_pred.round(), average='micro')
rmse = sqrt(mean_squared_error(y_test, y_pred))
print(f'Accuracy: {accuracy}')