diff options
author | TAMARA JERINIC <tamara.jerinic@gmail.com> | 2022-05-19 12:19:37 +0200 |
---|---|---|
committer | TAMARA JERINIC <tamara.jerinic@gmail.com> | 2022-05-19 12:19:37 +0200 |
commit | 7e1226a568a2c2a90f48a71f09828e909c23e266 (patch) | |
tree | 985dcd13798b542cc7e1dbefa509589d67ca384c /backend | |
parent | 3041334bd4686a8e0a9d434c2c99c13f39fa1ada (diff) |
Izmenjeno slanje metrika za regresiju.
Diffstat (limited to 'backend')
-rw-r--r-- | backend/microservice/api/newmlservice.py | 6 |
1 files changed, 2 insertions, 4 deletions
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))) |