机器学习sklearn:支持向量机svm

概述:现在就只知道这个svm可以画出决策边界,对数据的划分。简单举例就是:好的和坏的数据分开,中间的再验证

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

# 1. 生成数据
X, y = make_blobs(
    n_samples=50,       # 50个样本
    centers=2,          # 2个类别(二分类)
    random_state=0,     # 固定随机种子
    cluster_std=0.6     # 控制数据点的分散程度
)   # X是样本,y是标签

# 2. 训练线性SVM
model = SVC(kernel='linear').fit(X, y)

# 3. 绘制数据点
plt.scatter(X[:, 0], X[:, 1], c=y, s=50, cmap="rainbow")  # 按类别着色

# 4. 绘制决策边界和支持向量间隔
ax = plt.gca()  # 获取当前坐标轴
xlim = ax.get_xlim()  # 获取x轴范围
ylim = ax.get_ylim()  # 获取y轴范围

# 生成网格点(用于绘制决策边界)
xx, yy = np.meshgrid(
    np.linspace(xlim[0], xlim[1], 50),  # x轴50个点
    np.linspace(ylim[0], ylim[1], 50)   # y轴50个点
)   # 分成50分画网格

# 计算决策函数值(SVM的间隔)
Z = model.decision_function(np.c_[xx.ravel(), yy.ravel()]).reshape(xx.shape)    # 计算决策值并重塑为网格矩阵

# 绘制决策边界(黑色实线)和间隔边界(黑色虚线)
ax.contour(
    xx, yy, Z,
    colors='k',
    levels=[-1, 0, 1],  # 0是决策边界,±1是支持向量间隔
    alpha=0.5,          # 透明度
    linestyles=['--', '-', '--']  # 虚线-实线-虚线
)

# 5. 标记支持向量(SVM的关键数据点)
ax.scatter(
    model.support_vectors_[:, 0],  # 支持向量的x坐标
    model.support_vectors_[:, 1],  # 支持向量的y坐标
    s=100,                        # 点的大小
    facecolors='none',            # 空心点
    edgecolors='k',               # 黑色边框
    linewidths=1.5                # 边框宽度
)

plt.title("SVM Decision Boundary with Support Vectors")
plt.show()

print(Z)
相关推荐
西猫雷婶1 分钟前
神经网络|(十二)概率论基础知识-先验/后验/似然概率基本概念
人工智能·神经网络·机器学习·回归·概率论
Godspeed Zhao2 小时前
Tesla自动驾驶域控制器产品(AutoPilot HW)的系统化梳理
人工智能·机器学习·自动驾驶
fsnine2 小时前
机器学习案例——预测矿物类型(模型训练)
人工智能·机器学习
没有梦想的咸鱼185-1037-166311 小时前
AI大模型支持下的:CMIP6数据分析与可视化、降尺度技术与气候变化的区域影响、极端气候分析
人工智能·python·深度学习·机器学习·chatgpt·数据挖掘·数据分析
fanzhix15 小时前
线性回归学习
学习·机器学习·线性回归
OAFD.1 天前
机器学习之线性回归:原理、实现与实践
人工智能·机器学习·线性回归
Gloria_niki1 天前
机器学习之K 均值聚类算法
人工智能·机器学习
codeyanwu1 天前
nanoGPT 部署
python·深度学习·机器学习
Loving_enjoy1 天前
智能合约漏洞检测技术综述:守护区块链世界的“自动售货机”
经验分享·机器学习·课程设计·facebook
严文文-Chris1 天前
【大模型量化、蒸馏、剪枝、微调小结】
算法·机器学习·剪枝