决策树图绘制:Python 代码示例和可视化
决策树图绘制: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 库,我们可以轻松地构建和可视化决策树图。决策树图可以帮助我们直观地理解决策树的结构和预测过程。
注意:
本文提供的代码示例仅供参考,您可能需要根据您的实际情况进行调整。
原文地址: https://www.cveoy.top/t/topic/mrAD 著作权归作者所有。请勿转载和采集!