From b573b0069512f4444aefdac0a24eb99c8fcb6586 Mon Sep 17 00:00:00 2001 From: Danijel Andjelkovic Date: Thu, 31 Mar 2022 13:48:13 +0200 Subject: Malo sredio add-model stranicu, popravio bug sa prikazom javnih datasetova na naslovnoj strani. ML socket salje poruke za epohe backendu. --- backend/microservice/api/ml_service.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) (limited to 'backend/microservice/api/ml_service.py') diff --git a/backend/microservice/api/ml_service.py b/backend/microservice/api/ml_service.py index 68595a89..ea562212 100644 --- a/backend/microservice/api/ml_service.py +++ b/backend/microservice/api/ml_service.py @@ -124,17 +124,15 @@ def train(dataset, params, callback): # # Test # + model_name = params['_id'] + y_pred=classifier.predict(x_test) if(problem_type == "regresioni"): classifier.evaluate(x_test, y_test) - classifier.save("temp/"+model_name, save_format='h5') elif(problem_type == "binarni-klasifikacioni"): - y_pred=classifier.predict(x_test) y_pred=(y_pred>=0.5).astype('int') - #y_pred=(y_pred * 100).astype('int') - y_pred=y_pred.flatten() - result=pd.DataFrame({"Actual":y_test,"Predicted":y_pred}) - model_name = params['_id'] - classifier.save("temp/"+model_name, save_format='h5') + y_pred=y_pred.flatten() + result=pd.DataFrame({"Actual":y_test,"Predicted":y_pred}) + classifier.save("temp/"+model_name, save_format='h5') # # Metrike # -- cgit v1.2.3