update dllib-sacred.py

2022-05-09 10:53:49 +02:00 · 2022-05-09 10:53:49 +02:00 · 6cda242a4c
commit 6cda242a4c
parent ab7444f594
1 changed files with 1 additions and 0 deletions
--- a/Biblioteka_DL/dllib-sacred.py
+++ b/Biblioteka_DL/dllib-sacred.py
@ -323,6 +323,7 @@ def my_main(epochs, _run):

    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.log_scalar("measure.accuracy", accuracy_score(labels_test_g, np.argmax(pred, axis=1)))
    pred = pd.DataFrame(pred)

    pred.to_csv('result.csv')