决策树之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()
相关推荐
学嵌入式的小杨同学3 分钟前
循环队列(顺序存储)完整解析与实现(数据结构专栏版)
c语言·开发语言·数据结构·c++·算法
shangjian0074 分钟前
AI大模型-机器学习-算法-线性回归-优化方法
人工智能·算法·机器学习
shangjian0077 分钟前
AI大模型-机器学习-算法-逻辑回归
人工智能·算法·机器学习
王锋(oxwangfeng)9 分钟前
车道线拟合算法--自动驾驶
人工智能·算法·自动驾驶
computersciencer10 分钟前
一文快速理解线性回归的过程
算法·机器学习·回归·线性回归
VT.馒头14 分钟前
【力扣】2627. 函数防抖
前端·javascript·算法·leetcode
想逃离铁厂的老铁15 分钟前
Day41 >> 121、买卖股票的最佳时机 + 122.买卖股票的最佳时机II + 123.买卖股票的最佳时机III
算法·leetcode
夏鹏今天学习了吗15 分钟前
【LeetCode热题100(79/100)】打家劫舍
算法·leetcode·职场和发展
小欣加油25 分钟前
leetcode 面试题17.16 按摩师
数据结构·c++·算法·leetcode·动态规划
飞Link30 分钟前
数据合成中的通用模型蒸馏、领域模型蒸馏和模型自我提升
算法·数据挖掘