Python实现应用决策树的实例程序

决策树是一种常用的机器学习算法,用于分类和回归任务。以下是使用Python和scikit-learn库实现决策树分类器的简单实例程序,用于对鸢尾花(Iris)数据集进行分类。

首先,确保你已经安装了scikit-learn库。如果没有安装,可以通过以下命令安装:
登录后复制

plain 复制代码
pip install scikit-learn

以下是一个决策树分类器的实例程序:
登录后复制

plain 复制代码
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.tree import DecisionTreeClassifier, plot_tree  # 导入plot_tree函数
from sklearn.metrics import accuracy_score
import matplotlib.pyplot as plt

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

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

# 创建决策树分类器实例
clf = DecisionTreeClassifier(random_state=42)

# 训练模型
clf.fit(X_train, y_train)

# 预测测试集
y_pred = clf.predict(X_test)

# 计算准确率
accuracy = accuracy_score(y_test, y_pred)
print(f"Accuracy: {accuracy:.2f}")

# 可视化决策树
plt.figure(figsize=(12, 8))
plot_tree(clf, filled=True, feature_names=iris.feature_names, class_names=iris.target_names)
plt.title("Decision Tree for Iris Dataset")
plt.show()

在这个程序中,我们首先从scikit-learn库中加载了鸢尾花数据集,并将数据集划分为训练集和测试集。然后,我们创建了一个DecisionTreeClassifier实例,并使用训练集对其进行训练。接着,我们使用训练好的模型对测试集进行预测,并计算模型的准确率。最后,我们使用matplotlib库和tree.plot_tree函数可视化决策树。

这个程序展示了决策树算法在分类任务中的应用,通过可视化决策树,我们可以直观地了解模型是如何根据特征进行分类决策的。

运行成功截图:

相关推荐
天天进步2015几秒前
Python项目-基于Flask的个人博客系统设计与实现(2)
开发语言·python·flask
mNinGInG5 分钟前
c++练习
开发语言·c++·算法
牛马baby13 分钟前
Java高频面试之并发编程-02
java·开发语言·面试
纪元A梦30 分钟前
分布式锁算法——基于ZooKeeper的分布式锁全面解析
java·分布式·算法·zookeeper
liruiqiang0531 分钟前
循环神经网络 - 通用近似定理 & 图灵完备
人工智能·rnn·深度学习·神经网络·机器学习
Panesle1 小时前
广告推荐算法:COSMO算法与A9算法的对比
人工智能·算法·机器学习·推荐算法·广告推荐
月亮被咬碎成星星1 小时前
LeetCode[15]三数之和
数据结构·算法·leetcode
凉白开3381 小时前
Scala基础知识
开发语言·后端·scala
不要不开心了1 小时前
Scala内容
开发语言·pytorch·flask·scala·dash
2401_824256861 小时前
Scala的函数式编程
开发语言·后端·scala