使用SVM算法和使用kNN一样,要做数据标准化处理(涉及距离,需要统一量纲)。
(1)导入所需的模块和包
import numpy as np
import matplotlib.pyplot as plt
from sklearn import datasets
(2)加载并提取部分数据集
iris = datasets.load_iris()
X = iris.data
y = iris.target
X = X[y < 2, :2]
y = y[y < 2]
(3)绘制出提取的数据集的点
plt.scatter(X[y == 0, 0], X[y == 0, 1], color='r')
plt.scatter(X[y == 1, 0], X[y == 1, 1], color='b')
plt.show()
(4)数据集的归一化处理
from sklearn.preprocessing import St