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