s440058/ium_440058
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add eval script
All checks were successful
s440058-evaluation/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
piotr6789 2021-05-24 13:14:10 +02:00
parent 0bd1982754
commit e8fffb09d4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch-example-evaluate.py
View File

@ -48,7 +48,7 @@ model = LogisticRegressionModel(input_dim, output_dim)
pred = model(fTest) pred = model(fTest)
accuracy = accuracy_score(tTest, np.argmax(pred.detach().numpy(), axis = 1)) accuracy = accuracy_score(tTest, np.argmax(pred.detach().numpy(), axis = 1))
f1 = f1_score(tTest, np.argmax(pred.detach().numpy(), axis = 1), average = None) f1 = f1_score(tTest, np.argmax(pred.detach().numpy(), axis = 1), average = None)
rmse = mean_squared_error(tTest, np.argmax(pred.detach().numpy())) rmse = mean_squared_error(tTest, pred.detach().numpy())
print(f'Accuracy: {accuracy}') print(f'Accuracy: {accuracy}')
print(f'F1: {f1_score}') print(f'F1: {f1_score}')