【机器学习】什么是支持向量机?

什么是支持向量机?

支持向量机(SVM,Support Vector Machine)是一种强大的机器学习算法,常用于分类问题,也可以用于回归问题。它的核心思想是通过找到一个最佳的"超平面"来将不同类别的数据分开,并使得不同类别之间的间隔最大化。

通俗易懂的理解:

想象你有一片二维的平面,上面有两种不同的点:一种用红色标记,另一种用蓝色标记。你需要画一条直线(或者高维空间中的超平面)来将这两类点分开,且使得这条直线距离两类点的最近点尽可能远。支持向量机就是通过这种方法找到最佳的分隔线(或超平面)。

举个例子:

假设我们有两类动物的体重和身高数据:

  • 红色:老虎
  • 蓝色:猫

我们用身高和体重来区分这两类动物。

身高(cm) 体重(kg) 动物类型
50 2
60 3
70 10 老虎
80 12 老虎

支持向量机的目标:

  1. 找到一条分割线(在二维空间中为直线,在高维空间中为超平面),使得它将两类点(猫和老虎)分开。
  2. 使得这条分割线距离每类点的"最近点"尽可能远(这个最近的点称为"支持向量")。

核心概念:

  1. 支持向量:距离决策边界(分割线)最近的样本点,这些点决定了分割线的位置。
  2. 超平面:在高维空间中的"分割面",在二维空间中就是一条直线。

SVM的决策边界:

支持向量机通过找到一个最大间隔的超平面来区分不同的类别,最大化的间隔就是"支持向量机"的优势之一,它使得模型对新数据的预测更加稳定。

代码示例:

我们可以使用 scikit-learn 库来实现一个简单的支持向量机分类器,来处理上面的猫和老虎的例子。

python 复制代码
from sklearn import svm
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import rcParams

# 设置字体为支持中文的字体
rcParams['font.family'] = 'SimHei'
rcParams['axes.unicode_minus'] = False

# 数据:身高和体重
X = np.array([[50, 2], [60, 3], [70, 10], [80, 12]])
# 标签:猫=0,老虎=1
y = np.array([0, 0, 1, 1])

# 创建SVM分类器
clf = svm.SVC(kernel='linear')

# 训练SVM模型
clf.fit(X, y)

# 创建一个网格用于画决策边界
xx, yy = np.meshgrid(np.linspace(40, 90, 100), np.linspace(0, 15, 100))
Z = clf.decision_function(np.c_[xx.ravel(), yy.ravel()])
Z = Z.reshape(xx.shape)

# 绘制数据点
plt.scatter(X[:, 0], X[:, 1], c=y, cmap='autumn')

# 绘制决策边界
plt.contour(xx, yy, Z, levels=[0], linewidths=2, colors='black')

# 标题
plt.title("SVM分类:猫 vs 老虎")

# 显示图形
plt.xlabel('身高 (cm)')
plt.ylabel('体重 (kg)')
plt.show()

解释:

  • X: 是我们输入的数据,表示每个动物的身高和体重。
  • y: 是标签,0表示猫,1表示老虎。
  • svm.SVC(kernel='linear'): 创建一个线性SVM分类器,kernel='linear'表示我们用的是线性核(即我们假设数据可以通过一条直线分开)。
  • clf.fit(X, y): 训练SVM模型。
  • clf.decision_function: 用来获取决策边界(分割线)的值。
  • plt.contour: 绘制决策边界。

输出图像:

图中会显示数据点,并且有一条黑色的线,表示支持向量机的决策边界,分割了猫和老虎的两类数据。

总结:

  • SVM 是一个通过找到最大间隔超平面来进行分类的算法。
  • 它的优点是对高维数据和非线性问题表现非常好,特别是使用核方法时(例如,使用 RBF核可以处理复杂的非线性数据)。
  • 通过支持向量机,模型会利用支持向量来构建最优的决策边界,保证分类的准确性。
相关推荐
fancy1661661 小时前
力扣top100 矩阵置零
人工智能·算法·矩阵
元亓亓亓1 小时前
LeetCode热题100--240.搜索二维矩阵--中等
算法·leetcode·矩阵
奋斗者1号2 小时前
Docker 部署 - Crawl4AI 文档 (v0.5.x)
人工智能·爬虫·机器学习
明月看潮生2 小时前
青少年编程与数学 02-019 Rust 编程基础 09课题、流程控制
开发语言·算法·青少年编程·rust·编程与数学
oioihoii2 小时前
C++23 views::slide (P2442R1) 深入解析
linux·算法·c++23
NLP小讲堂3 小时前
LLaMA Factory 深度调参
人工智能·机器学习
yuhao__z3 小时前
代码随想录算法训练营第六十三天| 图论9—卡码网47. 参加科学大会,94. 城市间货物运输 I
算法·图论
June`3 小时前
专题三:穷举vs暴搜vs深搜vs回溯vs剪枝(全排列)决策树与递归实现详解
c++·算法·深度优先·剪枝
Alessio Micheli3 小时前
基于几何布朗运动的股价预测模型构建与分析
线性代数·机器学习·概率论
vlln3 小时前
适应性神经树:当深度学习遇上决策树的“生长法则”
人工智能·深度学习·算法·决策树·机器学习