diff options
Diffstat (limited to 'backend')
| -rw-r--r-- | backend/microservice/api/newmlservice.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/backend/microservice/api/newmlservice.py b/backend/microservice/api/newmlservice.py index 631837e5..3244e82f 100644 --- a/backend/microservice/api/newmlservice.py +++ b/backend/microservice/api/newmlservice.py @@ -155,7 +155,8 @@ def train(dataset, paramsModel,paramsExperiment,paramsDataset,callback): #print(data) for col in paramsExperiment["inputColumns"]: #print(col) - data[col]=dataset[col] + if(col!=paramsExperiment["outputColumn"]): + data[col]=dataset[col] output_column = paramsExperiment["outputColumn"] data[output_column] = dataset[output_column] #print(data) @@ -201,7 +202,7 @@ def train(dataset, paramsModel,paramsExperiment,paramsDataset,callback): ''' ### Enkodiranje encodings=paramsExperiment["encodings"] - datafront=dataset.copy() + #datafront=dataset.copy() #svekolone=datafront.columns #kategorijskekolone=datafront.select_dtypes(include=['object']).columns for kolonaEncoding in encodings: |