diff options
Diffstat (limited to 'backend')
| -rw-r--r-- | backend/microservice/api/newmlservice.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/backend/microservice/api/newmlservice.py b/backend/microservice/api/newmlservice.py index 9951c25f..604e4d3c 100644 --- a/backend/microservice/api/newmlservice.py +++ b/backend/microservice/api/newmlservice.py @@ -155,11 +155,15 @@ def train(dataset, paramsModel,paramsExperiment,paramsDataset,callback): data.pop(col) # ### Enkodiranje - encoding=paramsExperiment["encoding"] + encodings=paramsExperiment["encodings"] datafront=dataset.copy() svekolone=datafront.columns kategorijskekolone=datafront.select_dtypes(include=['object']).columns - for kolona in svekolone: + for kolonaEncoding in encodings: + + kolona = kolonaEncoding["columnName"] + encoding = kolonaEncoding["encoding"] + if(kolona in kategorijskekolone): if(encoding=='label'): encoder=LabelEncoder() |