Skip to content

Commit 38d3267

Browse files
committed
minor edit to AdaboostClassifierTuner
1 parent f0b6da6 commit 38d3267

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tune_classifier/ensemble_classifier.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ class AdaBoostClassifierTuner(BaseTuner):
118118
estimator_space: Iterable[Optional[object]] = (None, )
119119
n_estimators_space: Dict[str, Any] = field(default_factory=lambda: {"low":1, "high":200, "step":1, "log":True})
120120
learning_rate_space: Dict[str, Any] = field(default_factory=lambda: {"low":0.01, "high":1, "step":None, "log":True})
121-
algorithm_space: Iterable[str] = ("SAMME", "SAMME.R")
121+
algorithm_space: Iterable[str] = ("SAMME", )
122122
random_state_space: Dict[str, Any] = field(default_factory=lambda: {"low":1, "high":10000, "step":1, "log":True})
123123

124124
def sample_params(self, trial: Optional[Trial]=None) -> Dict[str, Any]:

0 commit comments

Comments
 (0)