def __init__(self, task: Task, scorer: Scorer, opt_logger: OptimizationLogger=VoidLogger(None)):
if task.task == "classification":
space = RandomForestOptimizer.Params.classification_space
model = ensemble.RandomForestClassifier()
else:
space = RandomForestOptimizer.Params.regression_space
model = ensemble.RandomForestRegressor()
super().__init__(model, task, space, scorer, opt_logger)
评论列表
文章目录