diff options
| author | TAMARA JERINIC <tamara.jerinic@gmail.com> | 2022-05-05 00:05:27 +0200 |
|---|---|---|
| committer | TAMARA JERINIC <tamara.jerinic@gmail.com> | 2022-05-05 00:05:27 +0200 |
| commit | 6f798869049aee68cf84f381317cc1ee46dbcce2 (patch) | |
| tree | 7a47437df552df6c4c53f910ebcd89389e648d86 /backend/microservice/api | |
| parent | 75f0fca5f9e7e74979f63d73741512ea2e58e9e6 (diff) | |
Promenjena lokacija columnType parametra.
Diffstat (limited to 'backend/microservice/api')
| -rw-r--r-- | backend/microservice/api/newmlservice.py | 11 |
1 files changed, 7 insertions, 4 deletions
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() |