adjust configs to advise 3 initial assumptions; add prefix for config…

…_light.yaml
ITMO-NSS-team · MorrisNein · May 21, 2024 · Dec 13, 2023 · Mar 26, 2024 · Dec 13, 2023
commit 238483f375a503fdb459ccd1fa2ecdb52fd49471
diff --git a/experiments/fedot_warm_start/config.yaml b/experiments/fedot_warm_start/config.yaml
@@ -14,4 +14,4 @@ assessor_params:
   n_neighbors: 5
 advisor_params:
   minimal_distance: 1
-  n_best_to_advise: 5
+  n_best_to_advise: 3
diff --git a/experiments/fedot_warm_start/config_debug.yaml b/experiments/fedot_warm_start/config_debug.yaml
@@ -14,4 +14,4 @@ assessor_params:
   n_neighbors: 2
 advisor_params:
   minimal_distance: 1
-  n_best_to_advise: 5
+  n_best_to_advise: 3
diff --git a/experiments/fedot_warm_start/config_light.yaml b/experiments/fedot_warm_start/config_light.yaml
@@ -1,6 +1,7 @@
 ---
 seed: 42
 tmpdir: '/var/essdata/tmp'
+save_dir_prefix: light_
 #data_settings:
 n_datasets: 16 # null for all available datasets
 test_size: 0.25
@@ -15,4 +16,4 @@ assessor_params:
   n_neighbors: 5
 advisor_params:
   minimal_distance: 1
-  n_best_to_advise: 5
+  n_best_to_advise: 3
diff --git a/experiments/fedot_warm_start/evaluation_config.yaml b/experiments/fedot_warm_start/evaluation_config.yaml
@@ -1,11 +1,10 @@
-n_folds: 1
 split_seed: 0
 collect_metrics:
   - f1
   - roc_auc
   - accuracy
   - neg_log_loss
   - precision
-baseline_model: 'xgboost'
+baseline_model: 'catboost'
 data_test_size: 0.25
 data_split_seed: 0