feat: change learning rate values in grid

EducationalTestingService · Jun 26, 2023 · 143ff09 · 143ff09
1 parent 60274f6
commit 143ff09
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/doc/run_experiment.rst b/doc/run_experiment.rst
@@ -1110,14 +1110,14 @@ GradientBoostingClassifier and GradientBoostingRegressor
 HistGradientBoostingClassifier
     .. code-block:: python
 
-       {'learning_rate': [0.1, 0.5, 1.0],
+       {'learning_rate': [0.01, 0.1, 1.0],
         'min_samples_leaf': [10, 20, 40]}
 
 HistGradientBoostingRegressor
     .. code-block:: python
 
        {'loss': ['squared_error', 'absolute_error', 'poisson'],
-        'learning_rate': [0.1, 0.5, 1.0],
+        'learning_rate': [0.01, 0.1, 1.0],
         'min_samples_leaf': [10, 20, 40]}
 
 HuberRegressor

diff --git a/skll/utils/constants.py b/skll/utils/constants.py
@@ -72,12 +72,12 @@
     GradientBoostingClassifier: {"max_depth": [1, 3, 5]},
     GradientBoostingRegressor: {"max_depth": [1, 3, 5]},
     HistGradientBoostingClassifier: {
-        "learning_rate": [0.1, 0.5, 1.0],
+        "learning_rate": [0.01, 0.1, 1.0],
         "min_samples_leaf": [10, 20, 40],
     },
     HistGradientBoostingRegressor: {
         "loss": ["squared_error", "absolute_error", "poisson"],
-        "learning_rate": [0.1, 0.5, 1.0],
+        "learning_rate": [0.01, 0.1, 1.0],
         "min_samples_leaf": [10, 20, 40],
     },
     HuberRegressor: {