Question

我很想绘制模型，但是代码错误，表明该模型尚未安装。但是我安装了模型。有人可以帮助我为什么我遇到此错误吗？

我的代码在下面；

model = BalancedRandomForestClassifier(n_estimators = 200, random_state = 0, max_depth=6)

model.fit(x_train, y_train)
y_pred_rf = model.predict(x_test)

dot_data = StringIO()
export_graphviz(model, out_file=dot_data,  
            filled=True, rounded=True,
            special_characters=True)
graph = pydotplus.graph_from_dot_data(dot_data.getvalue())  
Image(graph.create_png())

错误在下面；

---------------------------------------------------------------------------
NotFittedError                            Traceback (most recent call last)
<ipython-input-57-0036434b9b2c> in <module>
 16 export_graphviz(model, out_file=dot_data,  
 17             filled=True, rounded=True,
 ---> 18             special_characters=True)
 19 graph = pydotplus.graph_from_dot_data(dot_data.getvalue())
 20 Image(graph.create_png())

 /opt/anaconda/envs/env_python/lib/python3.6/site- 
 packages/sklearn/tree/export.py in export_graphviz(decision_tree, out_file, 
 max_depth, feature_names, class_names, label, filled, leaves_parallel, 
 impurity, node_ids, proportion, rotate, rounded, special_characters, 
 precision)
 754     """
 755 
 --> 756     check_is_fitted(decision_tree, 'tree_')
 757     own_file = False
 758     return_string = False

 /opt/anaconda/envs/env_python/lib/python3.6/site- 
 packages/sklearn/utils/validation.py in check_is_fitted(estimator, 
 attributes, msg, all_or_any)
 912 
 913     if not all_or_any([hasattr(estimator, attr) for attr in 
 attributes]):
 --> 914         raise NotFittedError(msg % {'name': 
 type(estimator).__name__})
 915 
 916 

 NotFittedError: This BalancedRandomForestClassifier instance is not fitted 
 yet. Call 'fit' with appropriate arguments before using this method.

Answer 1

export_graphviz需要一个树模型。因此，当这是一个整体时，您需要遍历树。 BalancedRandomForestClassifier公开了estimators_的此类用法。如Parthasarathy Subburaj所述，您可以循环其他estimators_并调用该函数。

这表示我建议您使用sklearn.tree.plot_tree(...)：https://scikit-learn.org/stable/modules/generated/sklearn.tree.plot_tree.html

这是一个纯matplotlib绘图助手，如果您只想获取图像表示而无需使用graphviz

，它将变得更加简单

NotFittedError：此BalancedRandomForestClassifier实例尚未安装。使用此方法之前，请使用适当的参数调用“ fit”

1 个答案: