SVM-代码实现

import numpy as np
import sklearn
from sklearn import svm

path = u'./iris.data'  # 分类数据文件路径

#定义一个转换函数：将字符串转为浮点型数据
def iris_type(s):
    it = {b'Iris-setosa': 0, b'Iris-versicolor': 1, b'Iris-virginica': 2}
    return it[s]
#读取数据
data = np.loadtxt(path, dtype=float, delimiter=',', converters={4: iris_type})
#将Iris分为训练集和测试集，其中4表示当前单个样本的参数维度为4
x, y = np.split(data, (4,), axis=1)
#随机划分训练集与测试集
x_train, x_test, y_train, y_test = sklearn.model_selection.train_test_split(x, y, random_state=1, train_size=0.6)
#获取SVM分类模型
model = svm.SVC(C=0.8, kernel='rbf', gamma=20, decision_function_shape='ovr')
#将训练数据喂给SVM分类模型
model.fit(x_train, y_train.ravel())

train_accuracy = model.score(x_train, y_train)
test_accuracy = model.score(x_test, y_test)
print ("训练准确率:" , train_accuracy )
print("测试准确率:" ,test_accuracy)

数据来源：https://archive.ics.uci.edu/ml/machine-learning-databases/iris/iris.data