决策树之scikit-learn

实例

python 复制代码
from sklearn.datasets import load_iris
from sklearn import tree
import matplotlib.pyplot as plt

# Load iris dataset
iris = load_iris()
X, y = iris.data, iris.target

# Fit the classifier
clf = tree.DecisionTreeClassifier()
clf = clf.fit(X, y)

# Plot the decision tree
plt.figure(figsize=(15, 10))
tree.plot_tree(clf)
plt.show()
相关推荐
宝贝儿好10 分钟前
【强化学习实战】第十一章:Gymnasium库的介绍和使用(1)、出租车游戏代码详解(Sarsa & Q learning)
人工智能·python·深度学习·算法·游戏·机器学习
weixin_458872613 小时前
东华复试OJ二刷复盘2
算法
Charlie_lll3 小时前
力扣解题-637. 二叉树的层平均值
算法·leetcode
爱淋雨的男人4 小时前
自动驾驶感知相关算法
人工智能·算法·自动驾驶
wen__xvn4 小时前
模拟题刷题3
java·数据结构·算法
滴滴答滴答答4 小时前
机考刷题之 6 LeetCode 169 多数元素
算法·leetcode·职场和发展
Neteen5 小时前
【数据结构-思维导图】第二章:线性表
数据结构·c++·算法
礼拜天没时间.5 小时前
力扣热题100实战 | 第25期:K个一组翻转链表——从两两交换到K路翻转的进阶之路
java·算法·leetcode·链表·递归·链表反转·k个一组翻转链表
Swift社区5 小时前
LeetCode 400 第 N 位数字
算法·leetcode·职场和发展
再难也得平5 小时前
力扣239. 滑动窗口最大值(Java解法)
算法·leetcode·职场和发展