diff options
| author | Sonja Galovic <galovicsonja@gmail.com> | 2022-04-13 22:44:06 +0200 |
|---|---|---|
| committer | Sonja Galovic <galovicsonja@gmail.com> | 2022-04-13 22:44:06 +0200 |
| commit | 5d856cf20177711074b473cc2d2133b85826fe63 (patch) | |
| tree | c7029d75af6d53e40bf806c51f1f4fcfc2136104 /backend/microservice/api/ml_service.py | |
| parent | 3c4bfb16f55aa51a50ee99327caca639e072cdab (diff) | |
| parent | 3cb8b60ed4e36f39187e64270479f472e08fb615 (diff) | |
Merge branch 'dev' of http://gitlab.pmf.kg.ac.rs/igrannonica/neuronstellar into dev
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() |