支持向量机，英文全称为Support Vecto Machines，简称SVM。它是一种非常优秀的分类模型，特别在小样本、非线性以及高维模式识别中有很好的表现。SVM是由Vapnik团队在1992年提出，最初被用来解决二分类问题，后来逐渐发展成为可以处理多分类问题的算法。
Python是一种简洁而强大的编程语言，它实现了众多机器学习算法的包，其中包括SVM。本文将介绍通过Python实现支持向量机算法的步骤。
一、准备数据
我们来构造一组简单的训练数据。创建一个示例数据集，其中x1表示身高，x2表示体重，y为类别标签（0或1）。
import numpy as np
import matplotlib.pyplot as plt
np.random.seed(7)
X_train = np.array([[167, 75], [182, 80], [176, 85], [156, 50], [173, 70], [183, 90], [178, 75], [156, 45],
[162, 55], [163, 50], [159, 45], [180, 85]])
y_train = np.array([0, 1, 1, 0, 0, 1, 0, 0, 0, 0, 0, 1])
plt.scatter(X_train[y_train == 0][:, 0], X_train[y_train == 0][:, 1], c='r', s=40, label='Male')
plt.scatter(X_train[y_train == 1][:, 0], X_train[y_train == 1][:, 1], c='b', s=40, label='Female')
plt.legend()
plt.xlabel('Height')
plt.ylabel('Weight')
plt.show()在这个数据集中，我们将人群分类为男性或女性。
二、选择分类器
接下来，我们要选择适用于这个问题的分类器，即SVM。SVM有许多变种，但是在这里，我们使用的是线性SVM。
我们来构造一个SVM模型：
from sklearn.svm import SVC svm = SVC(kernel='linear') svm.fit(X_train, y_train)
在这里，我们使用的是SVC类，指定kernel参数为linear，表明我们使用线性核。
三、绘制决策边界
我们想要知道模型的性能如何，因此我们可以在绘制出分类器的决策边界：
def plot_decision_boundary(model, ax=None):
if ax is None:
ax = plt.gca()
x_min, x_max = ax.get_xlim()
y_min, y_max = ax.get_ylim()
xx, yy = np.meshgrid(np.linspace(x_min, x_max, 100),
np.linspace(y_min, y_max, 100))
Z = model.predict(np.c_[xx.ravel(), yy.ravel()]).reshape(xx.shape)
ax.contourf(xx, yy, Z, alpha=0.2)
ax.contour(xx, yy, Z, colors='black', linewidths=0.5)
ax.set_xlim([x_min, x_max])
ax.set_ylim([y_min, y_max])
plt.scatter(X_train[y_train == 0][:, 0], X_train[y_train == 0][:, 1], c='r', s=40, label='Male')
plt.scatter(X_train[y_train == 1][:, 0], X_train[y_train == 1][:, 1], c='b', s=40, label='Female')
plot_decision_boundary(svm)
plt.legend()
plt.xlabel('Height')
plt.ylabel('Weight')
plt.show()运行结束后，可以看到绘制出了分类器的决策边界。
四、预测新数据
我们可以用训练好的模型对新的数据进行预测。
X_test = np.array([[166, 70], [185, 90], [170, 75]]) y_test = svm.predict(X_test) print(y_test)
在这里，我们使用predict函数对三个新数据样本进行预测。它将返回它们的类别。
结论
在这篇文章中，我们介绍了如何使用Python中的支持向量机算法。我们通过创建一个简单的训练数据集，并使用线性SVM构建了一个分类器。我们还绘制了分类器的决策边界，并使用模型来预测了新的数据样本。SVM在许多场合也是非常受欢迎的算法，可以在很多领域获得好的表现。如果你想在处理数据时，掌握更多机器学习的算法，那么SVM也是非常值得学习的。
以上是Python中的支持向量机算法实例的详细内容。更多信息请关注PHP中文网其他相关文章！
