aboutsummaryrefslogtreecommitdiff
path: root/backend/microservice/api/controller.py
diff options
context:
space:
mode:
Diffstat (limited to 'backend/microservice/api/controller.py')
-rw-r--r--backend/microservice/api/controller.py1
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'])