From 3c2a0be2a169f16702ff9648dc65ef9efbae07fc Mon Sep 17 00:00:00 2001 From: Nevena Bojovic Date: Mon, 18 Apr 2022 21:27:40 +0200 Subject: Enkodiranje reseno. --- backend/microservice/api/newmlservice.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'backend/microservice/api') 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() -- cgit v1.2.3