【发布时间】:2021-05-27 15:28:24
【问题描述】:
假设我已经定义了一个这样的回归器
tree = MultiOutputRegressor(DecisionTreeRegressor(random_state=0))
tree.fit(X_train, y_train)
现在我想做一个网格交叉验证来优化参数ccp_alpha(我不知道它是否是最好的优化参数,但我以它为例)。因此,我这样做:
alphas = np.arange(0,2,0.1)
pipe_tree = Pipeline(steps=[('scaler', scaler), ('pca', pca), ('tree', tree)], memory = 'tmp')
treeCV = GridSearchCV(pipe_tree, dict( pca__n_components=n_components, tree__ccp_alpha=alphas ), cv=5, scoring ='r2', n_jobs=-1)
start_time = time.time()
treeCV.fit(X_train, y_train)
问题是我拿这个问题:
ValueError: Invalid parameter ccp_alpha for estimator Pipeline(memory='tmp',
steps=[('scaler', StandardScaler()), ('pca', PCA()),
('tree',
MultiOutputRegressor(estimator=DecisionTreeRegressor(random_state=0)))]). Check the list of available parameters with `estimator.get_params().keys()`.
如果我使用命令tree.get_params().keys(),它会打印一个可能的参数列表以在我的模型中更改。我认为问题在于GridSearchCV() 命令中的这个tree__ccp_alpha=alphas。但无论我做什么改变,它都不起作用。
【问题讨论】:
标签: python python-3.x machine-learning scikit-learn regression