From 0bd19827547c1d2ea6041c7b16daa13b9cf118f4 Mon Sep 17 00:00:00 2001 From: piotr6789 Date: Mon, 24 May 2021 13:12:18 +0200 Subject: [PATCH] add eval script --- pytorch-example-evaluate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch-example-evaluate.py b/pytorch-example-evaluate.py index 7db5957..f6a012a 100644 --- a/pytorch-example-evaluate.py +++ b/pytorch-example-evaluate.py @@ -47,7 +47,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)) +f1 = f1_score(tTest, np.argmax(pred.detach().numpy(), axis = 1), average = None) rmse = mean_squared_error(tTest, np.argmax(pred.detach().numpy())) print(f'Accuracy: {accuracy}')