From c9d2571bf376beb458251067433c46a5b18d982a Mon Sep 17 00:00:00 2001 From: piotr6789 Date: Mon, 24 May 2021 12:56:21 +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 b5a50c4..e1307cf 100644 --- a/pytorch-example-evaluate.py +++ b/pytorch-example-evaluate.py @@ -41,7 +41,7 @@ output_dim = 1 model = LogisticRegressionModel(input_dim, output_dim) -pred = model(xTest) +pred = model(fTest) accuracy = accuracy_score(fTest, np.argmax(pred.detach(), axis = 1)) print(f'Accuracy: {accuracy}') \ No newline at end of file