diff options
| author | Nevena Bojovic <nenabojov@gmail.com> | 2022-04-13 21:44:32 +0200 |
|---|---|---|
| committer | Nevena Bojovic <nenabojov@gmail.com> | 2022-04-13 21:44:32 +0200 |
| commit | 8568f5eadf09ff9536aa19054a129ab4aec49991 (patch) | |
| tree | d0ef5278d5c763e5325b9af88401d01489959dff /backend/microservice/api/ml_service.py | |
| parent | ea53b083e0513ddcd1faa1a6de9c89c9671e4eb3 (diff) | |
Doradjen training zahtev.
Diffstat (limited to 'backend/microservice/api/ml_service.py')
| -rw-r--r-- | backend/microservice/api/ml_service.py | 2 |
1 files changed, 1 insertions, 1 deletions
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() |