Merge pull request #1294 from cta-observatory/fewer_trees

Reduce number of trees in RF regressors
cta-observatory · Oct 1, 2024 · d1cd930 · d1cd930
2 parents e4a5b7a + 057b198
commit d1cd930
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/lstchain/data/lstchain_lhfit_config.json b/lstchain/data/lstchain_lhfit_config.json
@@ -69,7 +69,7 @@
     "max_depth": 30,
     "min_samples_leaf": 10,
     "n_jobs": -1,
-    "n_estimators": 150,
+    "n_estimators": 50,
     "bootstrap": true,
     "criterion": "squared_error",
     "max_features": 1.0,
@@ -86,7 +86,7 @@
     "max_depth": 30,
     "min_samples_leaf": 10,
     "n_jobs": -1,
-    "n_estimators": 150,
+    "n_estimators": 50,
     "bootstrap": true,
     "criterion": "squared_error",
     "max_features": 1.0,

diff --git a/lstchain/data/lstchain_standard_config.json b/lstchain/data/lstchain_standard_config.json
@@ -73,7 +73,7 @@
     "max_depth": 30,
     "min_samples_leaf": 10,
     "n_jobs": -1,
-    "n_estimators": 150,
+    "n_estimators": 50,
     "bootstrap": true,
     "criterion": "squared_error",
     "max_features": 1.0,
@@ -90,7 +90,7 @@
     "max_depth": 30,
     "min_samples_leaf": 10,
     "n_jobs": -1,
-    "n_estimators": 150,
+    "n_estimators": 50,
     "bootstrap": true,
     "criterion": "squared_error",
     "max_features": 1.0,