'GridSearchCV' object has no attribute 'tree_' 使用 graphviz 时出错

Question

我正在使用 gridsearchCV 为决策树回归器设置参数，如下所示。

    clf = GridSearchCV(DecisionTreeRegressor(random_state=99),parameters,refit=True,cv=5)  # default is MSE
    clf.fit(x_train, y_train)

然后我想使用 Graphviz 将此输出传递给图表

    export_graphviz(clf, out_file='tree.dot',feature_names=df_temp.columns,leaves_parallel=True)
    subprocess.call(['dot', '-Tpdf', 'tree.dot', '-o' 'tree.pdf'])

但是我收到一条错误消息： 'GridSearchCV' 对象没有属性 'tree_'

我真的被这个问题困住了，需要一些帮助！

Answer 1

export_graphviz 需要 DecisionTreeRegressor，而不是 GridSearchCV；尝试使用 export_graphviz(clf.best_estimator_, ...).

'GridSearchCV' object has no attribute 'tree_' 使用 graphviz 时出错

'GridSearchCV' object has no attribute 'tree_' error using graphviz

python

graphviz

scikit-learn