决策树图绘制:Python 代码示例和可视化

决策树是一种常用的机器学习算法,它可以用于分类和回归问题。决策树的结构类似于树状图,每个节点代表一个特征,每个分支代表一个特征值,叶子节点代表预测结果。

代码示例:

from sklearn import tree
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from matplotlib import pyplot as plt

# 加载鸢尾花数据集
data = load_iris()
X = data.data
y = data.target

# 将数据集分成训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)

# 创建决策树模型
clf = tree.DecisionTreeClassifier()
clf = clf.fit(X_train, y_train)

# 绘制决策树图
plt.figure(figsize=(12, 8))
tree.plot_tree(clf, filled=True, rounded=True, feature_names=data.feature_names, class_names=data.target_names)
plt.show()

决策树图:

          feature 2
             |
        +----+-----+
     否  |         | 是
        |         |
        |         |
      feature 1   'label:': '是'
        |         |
    +---+---+     |
  否 |       | 是  |
    |       |     |
    |       |     |
  'label:': '否' 'label:': '是' |
                    |
                    |

解释:

  • 图中每个节点代表一个特征,例如 'feature 2' 和 'feature 1'。
  • 每个分支代表一个特征值,例如 '否' 和 '是'。
  • 叶节点代表预测结果,例如 'label: 否' 和 'label: 是'。

总结:

通过使用 Python 的 scikit-learn 库,我们可以轻松地构建和可视化决策树图。决策树图可以帮助我们直观地理解决策树的结构和预测过程。

注意:

本文提供的代码示例仅供参考,您可能需要根据您的实际情况进行调整。

决策树图绘制:Python 代码示例和可视化

原文地址: https://www.cveoy.top/t/topic/mrAD 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录