From 66c147bc3154e531cfc78591a7451d904122fc1f Mon Sep 17 00:00:00 2001 From: TAMARA JERINIC Date: Sat, 16 Apr 2022 21:52:40 +0200 Subject: Ispravljeno obaveštavanje backend-a o epohama. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- backend/microservice/api/controller.py | 18 ++++++++++++++---- 1 file changed, 14 insertions(+), 4 deletions(-) (limited to 'backend/microservice/api/controller.py') diff --git a/backend/microservice/api/controller.py b/backend/microservice/api/controller.py index 437690ee..f0f36907 100644 --- a/backend/microservice/api/controller.py +++ b/backend/microservice/api/controller.py @@ -1,4 +1,6 @@ +from cmath import log from dataclasses import dataclass +from distutils.command.upload import upload from gc import callbacks from xmlrpc.client import DateTime import flask @@ -31,16 +33,24 @@ class Predictor: class train_callback(tf.keras.callbacks.Callback): - def __init__(self, x_test, y_test): + def __init__(self, x_test, y_test,modelId): self.x_test = x_test self.y_test = y_test + self.modelId=modelId # def on_epoch_end(self, epoch, logs=None): - print(epoch) + #print('Evaluation: ', self.model.evaluate(self.x_test,self.y_test),"\n") + + #print(epoch) + + #print(logs) + #ml_socket.send(epoch) #file = request.files.get("file") url = config.api_url + "/Model/epoch" - requests.post(url, epoch).text + r=requests.post(url, json={"Stat":str(logs),"ModelId":str(self.modelId),"EpochNum":epoch}).text + + #print(r) #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']) @@ -63,7 +73,7 @@ def train(): url = config.api_url + "/file/h5" files = {'file': open(filepath, 'rb')} - r=requests.post(url, files=files) + r=requests.post(url, files=files,data={"uploaderId":paramsExperiment['uploaderId']}) fileId=r.text predictor = Predictor( _id = "", -- cgit v1.2.3