Python实现SVM支持向量机的示例代码

 更新时间:2023年02月06日 08:41:24   作者:梦想橡皮擦  
SVM 的目的是在数据集中找到一条最佳分隔超平面,使得在这个超平面两侧的数据分别属于不同的类别,且该超平面与最近的数据点之间的距离最大。本文将通过Python实现SVM支持向量机,感兴趣的可以了解一下

下面是使用 scikit-learn 库中的 SVM 模型的示例代码:

from sklearn import svm
from sklearn.datasets import make_classification

# generate some example data
X, y = make_classification(n_features=4, random_state=0)

# fit an SVM model to the data
clf = svm.SVC(kernel='linear')
clf.fit(X, y)

# predict new data
print(clf.predict([[0, 0, 0, 0]]))

什么是支持向量机

支持向量机(SVM),SVM 的目的是在数据集中找到一条最佳分隔超平面,使得在这个超平面两侧的数据分别属于不同的类别,且该超平面与最近的数据点之间的距离最大。

这些最近的数据点称为支持向量,因此该模型被称为支持向量机。

SVM 可以处理线性可分和线性不可分的情况,也可以通过使用核函数来处理非线性问题。

SVM 在解决分类问题时,通常是通过构造一个最大间隔分离超平面来实现的。

在回归问题中,SVM 通过构造一个回归超平面来实现预测。

SVM 的核心思想是利用数学方法将高维数据映射到低维空间,从而对不可分数据进行线性分类。

SVM 模型具有很好的泛化能力,可以解决高维数据的分类问题。

SVM 模型可以使用不同的核函数,如线性核、多项式核、高斯核等,这样可以解决非线性问题。此外,SVM 还可以处理大量特征数据,并具有高效的计算速度。

用 Python 实现 SVM

使用 scikit-learn 库实现 SVM 分类

import numpy as np
from sklearn import datasets
from sklearn.model_selection import train_test_split
from sklearn.svm import SVC
from sklearn.metrics import accuracy_score

# 加载数据集
iris = datasets.load_iris()
X = iris["data"]
y = iris["target"]

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=0)

# 创建 SVM 模型
model = SVC(kernel='linear', C=1, random_state=0)

# 训练模型
model.fit(X_train, y_train)

# 在测试集上进行预测
y_pred = model.predict(X_test)

# 计算预测精度
accuracy = accuracy_score(y_test, y_pred)
print("Accuracy:", accuracy)

首先使用 load_iris() 函数加载 iris 数据集,然后使用 train_test_split() 函数将数据集划分为训练集和测试集。

接着,使用 SVC 函数创建一个 SVM 模型,并使用 fit() 函数在训练集上进行训练。

最后,使用 predict 函数在测试集上进行预测,并使用 accuracy_score() 函数计算预测精度。

多项式和高斯核函数的 SVM

使用高斯核函数的 SVM 可以如下实现:

import numpy as np
from sklearn import datasets
from sklearn.model_selection import train_test_split
from sklearn.svm import SVC
from sklearn.metrics import accuracy_score

# 加载数据集
iris = datasets.load_iris()
X = iris["data"]
y = iris["target"]

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=0)

# 创建 SVM 模型,使用高斯核函数
model = SVC(kernel='rbf', C=1, random_state=0)

# 训练模型
model.fit(X_train, y_train)

# 在测试集上进行预测
y_pred = model.predict(X_test)

# 计算预测精度
accuracy = accuracy_score(y_test, y_pred)
print("Accuracy:", accuracy)

在上面的代码中,首先加载了 scikit-learn 中的 iris 数据集,并将其特征数据存储在 X 中,标签存储在 y 中。

使用 train_test_split() 函数划分出了训练集和测试集,并将其分别存储在 X_train , X_test , y_train , y_test 中。

创建了一个 SVM 模型,使用了高斯核函数(Radial basis function, RBF)。参数 C 用于控制对误差的惩罚程度,数值越大,惩罚程度越大。参数 gamma 控制高斯核函数的影响范围,数值越小,影响范围越广。

使用 fit() 方法训练模型。训练后,使用 predict() 方法在测试集上进行预测,并将预测结果存储在 y_pred 中。

最终的结果将通过 accuracy_score() 函数计算预测精度,并将其打印出来。

还可以调整 SVM 模型的超参数以改变模型的表现,例如 C 和 gamma 。通过调整超参数可以使模型更加适合数据,并提高模型的预测精度。

到此这篇关于Python实现SVM支持向量机的示例代码的文章就介绍到这了,更多相关Python SVM支持向量机内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • PyQt QDoubleSpinBox控件用法示例详解

    PyQt QDoubleSpinBox控件用法示例详解

    QDoubleSpinBox 是 PyQt中用于输入浮点数的控件,支持键盘输入和上下箭头调整数值,本文给大家介绍PyQt QDoubleSpinBox控件用法,感兴趣的朋友一起看看吧
    2025-04-04
  • python中property属性的介绍及其应用详解

    python中property属性的介绍及其应用详解

    这篇文章主要介绍了python中property属性的介绍及其应用详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-08-08
  • 利用python实现简单的循环购物车功能示例代码

    利用python实现简单的循环购物车功能示例代码

    购物车对我们每位开发者来说应该都不陌生,下面这篇文章主要给大家介绍了利用python实现简单的循环购物车功能的相关资料,文中给出了详细的示例代码供大家参考学习,需要的朋友们下面来一起看看吧。
    2017-07-07
  • 利用Python list列表修改元素

    利用Python list列表修改元素

    这篇文章主要介绍了利用Python list列表修改元素,Python 提供了两种修改列表(list)元素的方法,我们可以每次修改单个元素,也可以每次修改一组元素
    2022-06-06
  • python批量生成身份证号到Excel的两种方法实例

    python批量生成身份证号到Excel的两种方法实例

    这篇文章主要给大家介绍了关于python批量生成身份证号到Excel的两种方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2021-01-01
  • python configparser中默认值的设定方式

    python configparser中默认值的设定方式

    这篇文章主要介绍了python configparser中默认值的设定方式,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
    2022-02-02
  • 利用Python绘制MySQL数据图实现数据可视化

    利用Python绘制MySQL数据图实现数据可视化

    这篇文章主要介绍了利用Python绘制MySQL数据图实现数据可视化,其中包括Python与MySQL的连接搭建、用Python执行MySQL语句查询等内容,需要的朋友可以参考下
    2015-03-03
  • Python实现PC屏幕截图并自动发送邮件

    Python实现PC屏幕截图并自动发送邮件

    在当前的数字化世界中,自动化已经成为我们日常生活和工作中的关键部分,本文我们将探讨如何使用Python来实现一个特定的自动化任务 - PC屏幕截图自动发送到指定的邮箱,感兴趣的可以了解下
    2023-11-11
  • Python 数据结构之旋转链表

    Python 数据结构之旋转链表

    这篇文章主要介绍了Python 数据结构之旋转链表的相关资料,需要的朋友可以参考下
    2017-02-02
  • Pytorch 使用opnecv读入图像由HWC转为BCHW格式方式

    Pytorch 使用opnecv读入图像由HWC转为BCHW格式方式

    这篇文章主要介绍了Pytorch 使用opnecv读入图像由HWC转为BCHW格式方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-06-06

最新评论