决策树代码展示树
时间: 2023-08-07 17:20:16 浏览: 72
BigDataMining-决策树matlab代码
好的,展示决策树的代码如下(假设使用的是scikit-learn库中的决策树算法):
```python
from sklearn.tree import plot_tree
import matplotlib.pyplot as plt
# 假设已经训练好了一个决策树模型,命名为clf
plt.figure(figsize=(20, 10))
plot_tree(clf, filled=True)
plt.show()
```
这段代码会绘制出一个展示决策树结构的图形,并且树的每个节点的颜色代表了该节点所代表的类别,颜色越深代表概率越高。如果想要将这个图形保存下来,可以使用plt.savefig()函数。
阅读全文