#plot #tree #decision-tree #iris-dataset
Вопрос:
При реализации набора данных 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)
но все равно он не возвращает мое дерево со значениями классов.