This commit is contained in:
Dominik Strzako 2021-05-15 20:45:46 +02:00
parent 1189623cdb
commit e839c14f0c

View File

@ -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)