以下是实现决策树分类鸢尾花数据集的Python代码:

import pandas as pd
from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier, plot_tree
import matplotlib.pyplot as plt

# 加载鸢尾花数据集
iris = load_iris()

# 创建数据框
df = pd.DataFrame(iris.data, columns=iris.feature_names)
df['target'] = iris.target

# 创建决策树分类器
dtc = DecisionTreeClassifier(random_state=0)

# 训练模型
dtc.fit(df[iris.feature_names], df['target'])

# 可视化决策树
plt.figure(figsize=(20, 10))
plot_tree(dtc, feature_names=iris.feature_names, class_names=iris.target_names, filled=True)
plt.show()

运行以上代码,将输出一张决策树的可视化图表,如下所示:

决策树可视化图表

决策树的每个节点代表一个特征,分支代表特征的取值,最终的叶子节点代表分类的结果。通过观察决策树,可以看出分类鸢尾花的决策过程,例如,如果花瓣长度小于2.45,则分类为setosa,如果花瓣长度大于等于2.45且花瓣宽度小于等于1.75,则分类为versicolor,否则为virginica

写出决策树实现鸢尾花分类的python代码并实现可视化

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

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