from autoflow.pipeline.components.classification_base import AutoFlowClassificationAlgorithm
__all__ = ["AdaboostClassifier"]
[docs]class AdaboostClassifier(AutoFlowClassificationAlgorithm):
class__ = "AdaBoostClassifier"
module__ = "sklearn.ensemble"
tree_model = True
[docs] def after_process_hyperparams(self, hyperparams):
import sklearn.tree
hyperparams = super(AdaboostClassifier, self).after_process_hyperparams(hyperparams)
base_estimator = sklearn.tree.DecisionTreeClassifier(max_depth=hyperparams.pop("max_depth"))
hyperparams.update({
"base_estimator": base_estimator
})
return hyperparams