【Python】支持向量机SVM

示例代码:

python 复制代码
import numpy as np
import matplotlib.pyplot as plt
from sklearn import svm
from sklearn.datasets import make_blobs
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score, classification_report

# 设置中文字体支持
plt.rcParams['font.sans-serif'] = ['SimHei', 'Microsoft YaHei', 'KaiTi', 'SimSun']  # 使用系统自带字体
plt.rcParams['axes.unicode_minus'] = False  # 解决负号显示问题
# 1. 生成模拟数据
X, y = make_blobs(n_samples=100, centers=2,
                  random_state=6, cluster_std=1.2)
print("这是X")
print(X)
print("这是y")
print(y)
# 2. 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(
    X, y, test_size=0.2, random_state=42)

# 3. 创建SVM分类器(使用线性核)
clf = svm.SVC(kernel='linear', C=1.0)
clf.fit(X_train, y_train)

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

print(y_pred)
print(y_test)

# 5. 评估模型性能
print(f"测试集准确率: {accuracy_score(y_test, y_pred):.2f}")
print("\n分类报告:")
print(classification_report(y_test, y_pred))

# 6. 可视化结果
plt.figure(figsize=(10, 6))

# 绘制训练数据
plt.scatter(X_train[:, 0], X_train[:, 1], c=y_train,
            cmap='coolwarm', edgecolors='k', label='训练数据')

# 绘制测试数据
plt.scatter(X_test[:, 0], X_test[:, 1], c=y_test,
            cmap='coolwarm', marker='s', s=100,
            edgecolors='k', alpha=0.7, label='测试数据')

# 绘制决策边界
ax = plt.gca()
xlim = ax.get_xlim()
ylim = ax.get_ylim()

# 创建网格评估模型
xx = np.linspace(xlim[0], xlim[1], 30)
yy = np.linspace(ylim[0], ylim[1], 30)
YY, XX = np.meshgrid(yy, xx)
xy = np.vstack([XX.ravel(), YY.ravel()]).T
Z = clf.decision_function(xy).reshape(XX.shape)

# 绘制决策边界和间隔
ax.contour(XX, YY, Z, colors='k', levels=[-1, 0, 1],
           alpha=0.5, linestyles=['--', '-', '--'])
ax.scatter(clf.support_vectors_[:, 0], clf.support_vectors_[:, 1],
           s=100, linewidth=1, facecolors='none',
           edgecolors='k', label='支持向量')

plt.title('SVM分类结果可视化')
plt.xlabel('特征1')
plt.ylabel('特征2')
plt.legend()
plt.grid(True, linestyle='--', alpha=0.7)
plt.show()

可视化结果:

资料来源于:周志华-机器学习,如有侵权请联系删除

相关推荐
zandy101115 分钟前
LLM与数据工程的融合:衡石Data Agent的语义层与Agent框架设计
大数据·人工智能·算法·ai·智能体
计算机编程小央姐22 分钟前
数据安全成焦点:基于Hadoop+Spark的信用卡诈骗分析系统实战教程
大数据·hadoop·python·spark·毕业设计·课程设计·dash
max50060036 分钟前
本地部署开源数据生成器项目实战指南
开发语言·人工智能·python·深度学习·算法·开源
地平线开发者1 小时前
征程 6E/M|多 camera 场景示例
算法·自动驾驶
MonkeyKing_sunyuhua1 小时前
mac怎么安装uv工具
python·macos·uv
源代码•宸1 小时前
Leetcode—721. 账户合并【中等】
c++·经验分享·算法·leetcode·并查集
Source.Liu1 小时前
【Python基础】 19 Rust 与 Python if 语句对比笔记
笔记·python·rust
緈福的街口1 小时前
【leetcode】77.组合
算法·leetcode·职场和发展
NAGNIP1 小时前
深入 vLLM:高性能大模型推理框架解析
算法
JuneXcy1 小时前
字符串(2)
算法