@创建于:20210802
@修改于:20210802
- 使用scikitlearn和matplotlib的组合,可视化决策树非常简单。
tree.plot_tree(clf);
- dtreeviz方法
viz = dtreeviz(clf,
x_data=X_train,
y_data=y_train,
target_name=‘class’,
feature_names=iris.feature_names,
class_names=list(iris.target_names),
title=“Decision Tree - Iris data set”) viz
参考连接:
知乎:用dtreeviz实现决策树可视化