From 7cae50c68ae1c80ab34e118c6bb24a4a4cf99730 Mon Sep 17 00:00:00 2001 From: TAMARA JERINIC Date: Tue, 17 May 2022 23:06:55 +0200 Subject: Izmena regresije. --- backend/microservice/api/newmlservice.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'backend/microservice') diff --git a/backend/microservice/api/newmlservice.py b/backend/microservice/api/newmlservice.py index f374e9d2..afcbcdd7 100644 --- a/backend/microservice/api/newmlservice.py +++ b/backend/microservice/api/newmlservice.py @@ -370,7 +370,7 @@ def train(dataset, paramsModel,paramsExperiment,paramsDataset,callback): - classifier.compile(loss =paramsModel["lossFunction"] , optimizer =opt, metrics = ['accuracy','mae','mse']) + classifier.compile(loss =paramsModel["lossFunction"] , optimizer =opt, metrics = ['mae','mse']) history=classifier.fit( x=x_train, y=y_train, epochs = paramsModel['epochs'],batch_size=int(paramsModel['batchSize']),callbacks=callback(x_test, y_test,paramsModel['_id']),validation_data=(x_val, y_val)) -- cgit v1.2.3