diff options
Diffstat (limited to 'backend/microservice/api/controller.py')
-rw-r--r-- | backend/microservice/api/controller.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/backend/microservice/api/controller.py b/backend/microservice/api/controller.py index ceed02ad..059af317 100644 --- a/backend/microservice/api/controller.py +++ b/backend/microservice/api/controller.py @@ -16,6 +16,7 @@ class train_callback(tf.keras.callbacks.Callback): # def on_epoch_end(self, epoch, logs=None): print(epoch) + ml_socket.send(epoch) #print('Evaluation: ', self.model.evaluate(self.x_test,self.y_test),"\n") #broj parametara zavisi od izabranih metrika loss je default @app.route('/train', methods = ['POST']) |