s440058/ium_440058
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add eval script
Some checks failed
s440058-evaluation/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
piotr6789 2021-05-24 12:57:36 +02:00
parent c9d2571bf3
commit 2dd722ba58
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pytorch-example-evaluate.py
View File

@ -6,6 +6,7 @@ import numpy as np
import torch.nn.functional as F import torch.nn.functional as F
from torch.utils.data import DataLoader, TensorDataset, random_split from torch.utils.data import DataLoader, TensorDataset, random_split
from sklearn import preprocessing from sklearn import preprocessing
from sklearn.metrics import accuracy_score
import sys import sys
class LogisticRegressionModel(torch.nn.Module): class LogisticRegressionModel(torch.nn.Module):