From e839c14f0c9733bacdc5e252f59c1d052d7345eb Mon Sep 17 00:00:00 2001 From: Dominik Strzako Date: Sat, 15 May 2021 20:45:46 +0200 Subject: [PATCH] hotfix --- Zadanie_06_training.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/Zadanie_06_training.py b/Zadanie_06_training.py index 1a36a82..d928028 100644 --- a/Zadanie_06_training.py +++ b/Zadanie_06_training.py @@ -68,7 +68,13 @@ def prepare_model(epoch_param, batch_param, _run): NeuralModel.save('wine_model.h5') #TO TYLKO NA POTRZEBY ZADANIA - _run.info["Accuracy dla zbioru test (tylko na potrzeby zadania)"] = str(NeuralModel.predict(scaled_x)) + y_pred = NeuralModel.predict(scaled_x) + + y_pred = np.around(y_pred, decimals=0) + + results = accuracy_score(yenc,y_pred) + + _run.info["Accuracy dla zbioru test (tylko na potrzeby zadania)"] = str(results)