diff options
| author | Ivan Ljubisavljevic <ivan996sk@gmail.com> | 2022-05-18 02:51:55 +0200 |
|---|---|---|
| committer | Ivan Ljubisavljevic <ivan996sk@gmail.com> | 2022-05-18 02:51:55 +0200 |
| commit | 0a82c7181a54b48c010a4200ead1f469aff39497 (patch) | |
| tree | 5730a828edaf2a72c28248e3e318821a943de3e8 /backend/microservice/api/newmlservice.py | |
| parent | 7cae50c68ae1c80ab34e118c6bb24a4a4cf99730 (diff) | |
Sredjivanje baze i forme dataseta
Diffstat (limited to 'backend/microservice/api/newmlservice.py')
| -rw-r--r-- | backend/microservice/api/newmlservice.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/backend/microservice/api/newmlservice.py b/backend/microservice/api/newmlservice.py index afcbcdd7..fd21f8ce 100644 --- a/backend/microservice/api/newmlservice.py +++ b/backend/microservice/api/newmlservice.py @@ -510,7 +510,7 @@ def train(dataset, paramsModel,paramsExperiment,paramsDataset,callback): classifier.add(tf.keras.layers.Dense(units=1)) - classifier.compile(loss =paramsModel["lossFunction"] , optimizer = opt , metrics = ['mae','mse','rmse']) + 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)) hist=history.history |