diff options
| author | Nevena Bojovic <nenabojov@gmail.com> | 2022-04-01 21:42:09 +0200 |
|---|---|---|
| committer | Nevena Bojovic <nenabojov@gmail.com> | 2022-04-01 21:42:09 +0200 |
| commit | 70c96a90a12778e06499f7d30542afae137347a4 (patch) | |
| tree | 03a7660c1d2cdc76fee5543613234cf62df37258 /backend/microservice | |
| parent | b48cec261d204f39d7ea216b8abb3fdfa17fdad5 (diff) | |
Sitna korekcija.
Diffstat (limited to 'backend/microservice')
| -rw-r--r-- | backend/microservice/api/ml_service.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/backend/microservice/api/ml_service.py b/backend/microservice/api/ml_service.py index f1f34cb7..c7082454 100644 --- a/backend/microservice/api/ml_service.py +++ b/backend/microservice/api/ml_service.py @@ -134,10 +134,10 @@ def train(dataset, params, callback): test=params["randomTestSetDistribution"] randomOrder = params["randomOrder"] if(randomOrder): - random=50 + random=123 else: random=0 - x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=test, random_state=random) + x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=test, shuffle=params["shuffle"], random_state=random) # # Skaliranje vrednosti # @@ -173,8 +173,8 @@ def train(dataset, params, callback): model_name = params['_id'] #y_pred=classifier.predict(x_test) if(problem_type == "regresioni"): - print(classifier.evaluate(x_test, y_test)) y_pred=classifier.predict(x_test) + print(classifier.evaluate(x_test, y_test)) elif(problem_type == "binarni-klasifikacioni"): y_pred=classifier.predict(x_test) y_pred=(y_pred>=0.5).astype('int') |