diff options
| author | TAMARA JERINIC <tamara.jerinic@gmail.com> | 2022-05-12 09:04:05 +0200 |
|---|---|---|
| committer | TAMARA JERINIC <tamara.jerinic@gmail.com> | 2022-05-12 09:04:05 +0200 |
| commit | e8908768803c0fe037f5e873003078cc7cdb6293 (patch) | |
| tree | 57d474d05e585d8fd534db0fc6f6ed920a728a6d /backend/microservice | |
| parent | 1c6c681a151a5e7ad489b6b4ae7bc2073ceb1096 (diff) | |
Izmena parametara.
Diffstat (limited to 'backend/microservice')
| -rw-r--r-- | backend/microservice/api/newmlservice.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/backend/microservice/api/newmlservice.py b/backend/microservice/api/newmlservice.py index 07735077..c401a3e6 100644 --- a/backend/microservice/api/newmlservice.py +++ b/backend/microservice/api/newmlservice.py @@ -293,7 +293,7 @@ def train(dataset, paramsModel,paramsExperiment,paramsDataset,callback): random=0 #x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=test, random_state=random) #print(x_train,x_test) - x, x_test, y, y_test = train_test_split(x, y, test_size=0.15, shuffle=True) + x, x_test, y, y_test = train_test_split(x, y, test_size=test, random_state=random, shuffle=True) x_train, x_val, y_train, y_val = train_test_split(x, y, test_size=0.15, shuffle=True) # # Treniranje modela |