From 8568f5eadf09ff9536aa19054a129ab4aec49991 Mon Sep 17 00:00:00 2001 From: Nevena Bojovic Date: Wed, 13 Apr 2022 21:44:32 +0200 Subject: Doradjen training zahtev. --- backend/microservice/api/ml_service.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'backend/microservice/api/ml_service.py') diff --git a/backend/microservice/api/ml_service.py b/backend/microservice/api/ml_service.py index 4d2212f7..16ee7cc6 100644 --- a/backend/microservice/api/ml_service.py +++ b/backend/microservice/api/ml_service.py @@ -101,7 +101,7 @@ class TrainingResultRegression: class TrainingResult: metrics: dict ''' -def train(dataset, paramsModel, paramsExperiment, callback): +def train(dataset, paramsModel, paramsExperiment, paramsDataset, callback): problem_type = paramsModel["type"] dataModel = pd.DataFrame() dataExperiment = pd.DataFrame() -- cgit v1.2.3