s444356/ium_444356
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update dllib-sacred.py
All checks were successful
s444356-evaluation/pipeline/head This commit looks good
Details
s444356-training/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Maciej Czajka 2022-05-09 10:53:49 +02:00
parent ab7444f594
commit 6cda242a4c
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Biblioteka_DL/dllib-sacred.py
View File

@ -323,6 +323,7 @@ def my_main(epochs, _run):
print("The accuracy is", accuracy_score(labels_test_g, np.argmax(pred, axis=1))) print("The accuracy is", accuracy_score(labels_test_g, np.argmax(pred, axis=1)))
_run.info['accuracy'] = accuracy_score(labels_test_g, np.argmax(pred, axis=1)) _run.info['accuracy'] = accuracy_score(labels_test_g, np.argmax(pred, axis=1))
_run.log_scalar("measure.accuracy", accuracy_score(labels_test_g, np.argmax(pred, axis=1)))
pred = pd.DataFrame(pred) pred = pd.DataFrame(pred)
pred.to_csv('result.csv') pred.to_csv('result.csv')