From 7e1226a568a2c2a90f48a71f09828e909c23e266 Mon Sep 17 00:00:00 2001 From: TAMARA JERINIC Date: Thu, 19 May 2022 12:19:37 +0200 Subject: Izmenjeno slanje metrika za regresiju. --- backend/microservice/api/newmlservice.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) (limited to 'backend/microservice/api') diff --git a/backend/microservice/api/newmlservice.py b/backend/microservice/api/newmlservice.py index 427111a7..bcff5a33 100644 --- a/backend/microservice/api/newmlservice.py +++ b/backend/microservice/api/newmlservice.py @@ -514,15 +514,13 @@ def train(dataset, paramsModel,paramsExperiment,paramsDataset,callback): 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)) hist=history.history - print() + y_pred=classifier.predict(x_test) #print(classifier.evaluate(x_test, y_test)) classifier.save(filepath, save_format='h5') - metrics={} - + mse = float(sm.mean_squared_error(y_test,y_pred)) - metrics.append mae = float(sm.mean_absolute_error(y_test,y_pred)) mape = float(sm.mean_absolute_percentage_error(y_test,y_pred)) rmse = float(np.sqrt(sm.mean_squared_error(y_test,y_pred))) -- cgit v1.2.3