用jupyter notebook画一棵树

以下是使用Jupyter Notebook画一棵树的步骤：

1.首先，确保你已经安装了matplotlib库。如果没有安装，可以使用以下命令在Jupyter Notebook中安装：

!pip install matplotlib

2.导入matplotlib库和tree模块：

import matplotlib.pyplot as plt
from sklearn.tree import plot_tree

3.创建一个决策树模型并训练它：

from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier

iris = load_iris()
X = iris.data[:, 2:]
y = iris.target

tree_clf = DecisionTreeClassifier(max_depth=2)
tree_clf.fit(X, y)

4.使用plot_tree函数绘制决策树：

plt.figure(figsize=(10, 8))
plot_tree(tree_clf, filled=True)
plt.show()

这将在Jupyter Notebook中显示一棵决策树。