diff options
Diffstat (limited to 'backend/microservice/api/newmlservice.py')
| -rw-r--r-- | backend/microservice/api/newmlservice.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/backend/microservice/api/newmlservice.py b/backend/microservice/api/newmlservice.py index e81192ab..585db480 100644 --- a/backend/microservice/api/newmlservice.py +++ b/backend/microservice/api/newmlservice.py @@ -122,7 +122,7 @@ def train(dataset, paramsModel,paramsExperiment,paramsDataset,callback): for col in paramsExperiment["inputColumns"]: #print(col) data[col]=dataset[col] - output_column = paramsExperiment["columnToPredict"] + output_column = paramsExperiment["outputColumn"] data[output_column] = dataset[output_column] #print(data) @@ -297,8 +297,8 @@ def train(dataset, paramsModel,paramsExperiment,paramsDataset,callback): history=classifier.fit(x_train, y_train, epochs = paramsModel['epochs'],batch_size=paramsModel['batchSize']) hist=history.history - plt.plot(hist['accuracy']) - plt.show() + #plt.plot(hist['accuracy']) + #plt.show() y_pred=classifier.predict(x_test) y_pred=np.argmax(y_pred,axis=1) |