From 2dd722ba589ac3254dfb08fb3bf0f3e8b23b2d71 Mon Sep 17 00:00:00 2001
From: piotr6789 <piotr.ulanicki4@gmail.com>
Date: Mon, 24 May 2021 12:57:36 +0200
Subject: [PATCH] add eval script

---
 pytorch-example-evaluate.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pytorch-example-evaluate.py b/pytorch-example-evaluate.py
index e1307cf..fb68af5 100644
--- a/pytorch-example-evaluate.py
+++ b/pytorch-example-evaluate.py
@@ -6,6 +6,7 @@ import numpy as np
 import torch.nn.functional as F
 from torch.utils.data import DataLoader, TensorDataset, random_split
 from sklearn import preprocessing
+from sklearn.metrics import accuracy_score
 import sys
 
 class LogisticRegressionModel(torch.nn.Module):