plot - plot_tree 不返回 class_names

Question

在实现 iris 数据集时，我想在树形图中显示类名，但我无法弄清楚为什么它没有显示：

linear_regression = LinearRegression()
tree_reg = DecisionTreeRegressor(max_depth=3)

lr= linear_regression.fit(X_train, Y_train)
tr=tree_reg.fit(X_train, Y_train)

plt.figure(figsize=(10,15))
plot_tree(tr,feature_names=iris.feature_names, class_names=iris.target_names ,filled=True)

但是，它仍然没有返回带有类值的树。