from autoflow.pipeline.components.regression_base import AutoFlowRegressionAlgorithm
__all__ = ["AdaboostRegressor"]
[docs]class AdaboostRegressor(AutoFlowRegressionAlgorithm):
class__ = "AdaBoostRegressor"
module__ = "sklearn.ensemble"
[docs] def after_process_hyperparams(self, hyperparams):
import sklearn.tree
hyperparams = super(AdaboostRegressor, self).after_process_hyperparams(hyperparams)
base_estimator = sklearn.tree.DecisionTreeClassifier(max_depth=hyperparams.pop("max_depth"))
hyperparams.update({
"base_estimator": base_estimator
})
return hyperparams