＆＃39; GridSearchCV＆＃39;对象没有属性＆＃39; tree _＆＃39;使用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'])

但是我收到了错误消息： ＆＃39; GridSearchCV＆＃39;对象没有属性＆＃39; tree _＆＃39;

我真的坚持这一点，并且可以帮助你！

Answer 1

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