add eval script

2021-05-24 13:14:10 +02:00 · 2021-05-24 13:14:10 +02:00 · e8fffb09d4
commit e8fffb09d4
parent 0bd1982754
1 changed files with 1 additions and 1 deletions
--- a/pytorch-example-evaluate.py
+++ b/pytorch-example-evaluate.py
@ -48,7 +48,7 @@ model = LogisticRegressionModel(input_dim, output_dim)
 pred = model(fTest)
 accuracy = accuracy_score(tTest, np.argmax(pred.detach().numpy(), axis = 1))
 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'F1: {f1_score}')