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:56:21 +02:00
parent d76512c266
commit c9d2571bf3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch-example-evaluate.py
View File

@ -41,7 +41,7 @@ output_dim = 1
model = LogisticRegressionModel(input_dim, output_dim) model = LogisticRegressionModel(input_dim, output_dim)
pred = model(xTest) pred = model(fTest)
accuracy = accuracy_score(fTest, np.argmax(pred.detach(), axis = 1)) accuracy = accuracy_score(fTest, np.argmax(pred.detach(), axis = 1))
print(f'Accuracy: {accuracy}') print(f'Accuracy: {accuracy}')