From 6f798869049aee68cf84f381317cc1ee46dbcce2 Mon Sep 17 00:00:00 2001 From: TAMARA JERINIC Date: Thu, 5 May 2022 00:05:27 +0200 Subject: Promenjena lokacija columnType parametra. --- backend/microservice/api/newmlservice.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) (limited to 'backend/microservice/api') diff --git a/backend/microservice/api/newmlservice.py b/backend/microservice/api/newmlservice.py index d84d9567..826ac7cb 100644 --- a/backend/microservice/api/newmlservice.py +++ b/backend/microservice/api/newmlservice.py @@ -180,17 +180,20 @@ def train(dataset, paramsModel,paramsExperiment,paramsDataset,callback): kategorijskekolone=[] ###PRETVARANJE NUMERICKIH U KATREGORIJSKE AKO JE KORISNIK TAKO OZNACIO columnInfo=paramsDataset['columnInfo'] - for col in columnInfo: - if(col['columnType']=="Kategorijski"): + columnTypes=paramsExperiment['columnTypes'] + for i in range(len(columnInfo)): + col=columnInfo[i] + if(columnTypes[i]=="categorical"): data[col['columnName']]=data[col['columnName']].apply(str) kategorijskekolone.append(col['coumnName']) - + #kategorijskekolone=data.select_dtypes(include=['object']).columns + print(kategorijskekolone) ###NULL null_value_options = paramsExperiment["nullValues"] null_values_replacers = paramsExperiment["nullValuesReplacers"] if(null_value_options=='replace'): - #print("replace null") # + #print("replace null") dict=null_values_replacers while(len(dict)>0): replace=dict.pop() -- cgit v1.2.3