diff options
Diffstat (limited to 'backend/microservice/api/controller.py')
| -rw-r--r-- | backend/microservice/api/controller.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/backend/microservice/api/controller.py b/backend/microservice/api/controller.py index 73f4ed31..7852b63d 100644 --- a/backend/microservice/api/controller.py +++ b/backend/microservice/api/controller.py @@ -69,7 +69,7 @@ def train(): #dataset, paramsModel, paramsExperiment, callback) - filepath,result = newmlservice.train(data, paramsModel, paramsExperiment,paramsDataset, train_callback) + filepath,result,finalMetrics= newmlservice.train(data, paramsModel, paramsExperiment,paramsDataset, train_callback) """ f = request.json['filepath'] dataset = pd.read_csv(f) @@ -77,6 +77,7 @@ def train(): print(result) """ + url = config.api_url + "/file/h5" files = {'file': open(filepath, 'rb')} r=requests.post(url, files=files,data={"uploaderId":paramsExperiment['uploaderId']}) @@ -94,7 +95,9 @@ def train(): "experimentId" : paramsExperiment["_id"], "modelId" : paramsModel["_id"], "h5FileId" : fileId, - "metrics" : m + "metrics" : m, + "finalMetrics":finalMetrics + } #print(predictor) url = config.api_url + "/Predictor/add" |