Merge pull request #961 from cta-observatory/rfstddepth

Reduce size of random forests in standard config
cta-observatory · Apr 12, 2022 · 33a6355 · 33a6355
2 parents fb04c85 + 1289242
commit 33a6355
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 9 deletions.
diff --git a/lstchain/data/lstchain_standard_config.json b/lstchain/data/lstchain_standard_config.json
@@ -66,7 +66,7 @@
   },
 
   "random_forest_energy_regressor_args": {
-    "max_depth": 50,
+    "max_depth": 30,
     "min_samples_leaf": 2,
     "n_jobs": 4,
     "n_estimators": 150,
@@ -75,7 +75,7 @@
     "max_features": "auto",
     "max_leaf_nodes": null,
     "min_impurity_decrease": 0.0,
-    "min_samples_split": 2,
+    "min_samples_split": 10,
     "min_weight_fraction_leaf": 0.0,
     "oob_score": false,
     "random_state": 42,
@@ -84,7 +84,7 @@
   },
 
   "random_forest_disp_regressor_args": {
-    "max_depth": 50,
+    "max_depth": 30,
     "min_samples_leaf": 2,
     "n_jobs": 4,
     "n_estimators": 150,
@@ -93,7 +93,7 @@
     "max_features": "auto",
     "max_leaf_nodes": null,
     "min_impurity_decrease": 0.0,
-    "min_samples_split": 2,
+    "min_samples_split": 10,
     "min_weight_fraction_leaf": 0.0,
     "oob_score": false,
     "random_state": 42,
@@ -102,12 +102,12 @@
   },
 
   "random_forest_disp_classifier_args": {
-    "max_depth": 100,
+    "max_depth": 30,
     "min_samples_leaf": 2,
     "n_jobs": 4,
     "n_estimators": 100,
     "criterion": "gini",
-    "min_samples_split": 2,
+    "min_samples_split": 10,
     "min_weight_fraction_leaf": 0.0,
     "max_features": "auto",
     "max_leaf_nodes": null,
@@ -121,12 +121,12 @@
   },
 
   "random_forest_particle_classifier_args": {
-    "max_depth": 100,
+    "max_depth": 30,
     "min_samples_leaf": 2,
     "n_jobs": 4,
     "n_estimators": 100,
     "criterion": "gini",
-    "min_samples_split": 2,
+    "min_samples_split": 10,
     "min_weight_fraction_leaf": 0.0,
     "max_features": "auto",
     "max_leaf_nodes": null,

diff --git a/lstchain/tools/tests/test_tools.py b/lstchain/tools/tests/test_tools.py
@@ -139,7 +139,9 @@ def test_create_irf_point_like_energy_dependent_cuts(
                 "--overwrite",
                 "--energy-dependent-gh",
                 "--point-like",
-                "--energy-dependent-theta"
+                "--energy-dependent-theta",
+                "--DL3Cuts.max_theta_cut=1",
+                "--DL3Cuts.fill_theta_cut=1"
             ],
             cwd=temp_dir_observed_files,
         )