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)