如何显示此回归树？

Question

我已经在 jupyter notebook 中编写了回归树的代码，但仅此代码不会显示输出。我会使用 graphviz，但使用 graphviz 的所有方法也需要 qeds，我的机器拒绝安装。有没有什么方法可以在没有 graphviz 或 qeds 的情况下显示这棵树？

import seaborn as sns
from sklearn.model_selection import train_test_split
from sklearn.tree import DecisionTreeRegressor
from sklearn.metrics import r2_score,mean_squared_error

x = maindf["Graduate Degree"].values.reshape(-1, 1)
y = maindf["Democrats 2016"].values.reshape(-1, 1)
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.30, random_state=42)

DecisionTreeRegModel = DecisionTreeRegressor(max_depth=3)
DecisionTreeRegModel.fit(x_train,y_train)
y_pred = DecisionTreeRegModel.predict(x_test)

Answer 1

可以使用 scikit-learn 的 tree.plot_tree 来可视化树。

from sklearn import tree

_ = tree.plot_tree(DecisionTreeRegModel)

如何显示此回归树？

How do I display this regression tree?

python

graphviz

scikit-learn

jupyter-notebook