s151636/ium_151636
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

upd scr

Browse Source
This commit is contained in:
eugene 2023-06-06 22:38:32 +02:00
parent 1c183a567a
commit c1271fb458
1 changed files with 1 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
script5_4.py
View File

@ -4,7 +4,7 @@ import torch.nn as nn
from torch.utils.data import DataLoader, Dataset
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import LabelEncoder
from sklearn.metrics import accuracy_score, precision_score, recall_score, f1_score, mean_squared_error
from sklearn.metrics import mean_squared_error
import pickle
# Define the neural network model
@ -75,14 +75,6 @@ for inputs, targets_batch in test_dataloader:
predictions = torch.FloatTensor(predictions).squeeze()
targets = torch.FloatTensor(targets).squeeze()
accuracy = accuracy_score(targets, torch.round(predictions))
precision = precision_score(targets, torch.round(predictions), average='micro')
recall = recall_score(targets, torch.round(predictions), average='micro')
f1 = f1_score(targets, torch.round(predictions), average='micro')
rmse = mean_squared_error(targets, predictions, squared=False)
print("Accuracy:", accuracy)
print("Micro-average Precision:", precision)
print("Micro-average Recall:", recall)
print("F1 Score:", f1)
print("RMSE:", rmse)