From 72c9226b74f20e7d7d80e8267b8bc6a108a87018 Mon Sep 17 00:00:00 2001 From: aimspot Date: Tue, 28 Nov 2023 23:29:23 +0200 Subject: [PATCH] fix: ml model split data --- ODRS/ml_utils/ml_model_optimizer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ODRS/ml_utils/ml_model_optimizer.py b/ODRS/ml_utils/ml_model_optimizer.py index a5c0696..be148bd 100755 --- a/ODRS/ml_utils/ml_model_optimizer.py +++ b/ODRS/ml_utils/ml_model_optimizer.py @@ -27,8 +27,6 @@ def predict(mode, classes_path, dataset_path, speed, accuracy): file = Path(__file__).resolve() - #split_data(dataset_path, split_train_value=0.6, split_valid_value=0.35) - run_path = createRunDirectory(model='ml') model_top = list() @@ -37,6 +35,8 @@ def predict(mode, classes_path, dataset_path, speed, accuracy): dataset_path_new = getDataPath(ROOT, dataset_path) + split_data(dataset_path_new, split_train_value=0.6, split_valid_value=0.35) + dataset_data = dataset_info(dataset_path_new, Path(file.parents[2]) / classes_path, run_path) features_normalized, labels = dataProcessing(dataset_data, mode, speed, accuracy)