怎么在Jupyter Notebook中导出决策树已经画出的图为pdf
时间: 2024-02-25 12:55:54 浏览: 92
要在 Jupyter Notebook 中导出已经画出的决策树图为 PDF 文件,可以使用 Graphviz 库和 Pydotplus 库来实现。具体的代码如下:
首先,在 Jupyter Notebook 中安装必要的库:
```python
!pip install graphviz
!pip install pydotplus
```
然后,用下列代码将已经画好的决策树导出为 PDF 文件:
```python
from IPython.display import Image
from sklearn.tree import export_graphviz
import pydotplus
from sklearn.externals.six import StringIO
dot_data = StringIO()
export_graphviz(clf, out_file=dot_data,
filled=True, rounded=True,
special_characters=True, feature_names = feature_cols,class_names=['0','1'])
graph = pydotplus.graph_from_dot_data(dot_data.getvalue())
graph.write_pdf("decision_tree.pdf")
```
其中,clf 是已经训练好的决策树模型,feature_cols 是特征列的名称,class_names 是分类的名称。执行完这段代码后,当前目录下就会生成名为 decision_tree.pdf 的文件,即为导出的决策树图。
阅读全文