fix: ml model split data

aimclub · Nov 28, 2023 · 72c9226 · 72c9226
1 parent 5b164e7
commit 72c9226
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)