SVM的sklearn實現

轉載：豆-Metcalf

1）SVM-LinearSVC.ipynb-線性分類SVM，iris數據集分類，正確率100%

"""
功能：實現線性分類支持向量機
說明：可以用於二類分類，也可以用於多類分類
作者：唐天澤
博客：http://write.blog.csdn.net/mdeditor#!postId=76188190
日期：2017-08-09
"""

# 導入本項目所需要的包
import pandas as pd
import numpy as np

from sklearn import datasets

from sklearn import svm

# 使用交叉驗證的方法，把數據集分為訓練集合測試集
from sklearn.model_selection import train_test_split

# 加載iris數據集
def load_data():
    iris = datasets.load_iris()
    """展示數據集的形狀
       diabetes.data.shape, diabetes.target.shape
    """

    # 將數據集拆分為訓練集和測試集 
    X_train, X_test, y_train, y_test = train_test_split(
    iris.data, iris.target, test_size=0.10, random_state=0)
    return X_train, X_test, y_train, y_test

# 使用LinearSVC考察線性分類SVM的預測能力
def test_LinearSVC(X_train,X_test,y_train,y_test):

    # 選擇模型
    cls = svm.LinearSVC()

    # 把數據交給模型訓練
    cls.fit(X_train,y_train)

    print('Coefficients:%s, intercept %s'%(cls.coef_,cls.intercept_))
    print('Score: %.2f' % cls.score(X_test, y_test))

if __name__=="__main__":
    X_train,X_test,y_train,y_test=load_data() # 生成用於分類的數據集
    test_LinearSVC(X_train,X_test,y_train,y_test) # 調用 test_LinearSVC

 

2) SVM-LinearSVC-kaggle.ipynb-線性分類SVM，手寫數字數據集分類，正確率85%

"""
功能：實現線性分類支持向量機
說明：可以用於二類分類，也可以用於多類分類
作者：唐天澤
博客：http://write.blog.csdn.net/mdeditor#!postId=76188190
日期：2017-08-09
"""

# 導入本項目所需要的包
import pandas as pd
import numpy as np

from sklearn import datasets

from sklearn import svm

# 使用交叉驗證的方法，把數據集分為訓練集合測試集
from sklearn.model_selection import train_test_split

# The competition datafiles are in the directory ../input
# 加載數據集
def load_data():
    dataset = pd.read_csv("~/Desktop/knn/input/train.csv")
    label = dataset.values[:,0]
    train = dataset.values[:,1:]
    testdata  = pd.read_csv("~/Desktop/knn/input/test.csv").values
    return label,train,testdata
 # 使用LinearSVC考察線性分類SVM的預測能力
def test_LinearSVC(label,train,testdata):

    # 選擇模型
    cls = svm.LinearSVC()

    # 把數據交給模型訓練
    cls.fit(train,label)

    # 預測數據
    #print(cls.predict(testdata))
    results=cls.predict(testdata)
    return results

if __name__=="__main__":
    label,train,testdata = load_data()
    result = test_LinearSVC(label,train,testdata)
    pd.DataFrame({"ImageId": list(range(1,len(testdata)+1)),"Label": result}).to_csv(
    '~/Desktop/knn/output/Digit_recogniser_SVM_LinearSVC.csv', index=False,header=True)

 

補充：

from sklearn import  svm

from sklearn.datasets import  load_iris

from sklearn.model_selection import train_test_split

datas = load_iris()
# print(datas)
data_x = datas.data
data_y = datas.target
# print(data_x)

#print(data_y)

x_train,x_test,y_train,y_test = train_test_split(data_x,data_y,test_size=0.3)

clf = svm.SVC()#默認核函數是高斯核
# print(clf)
clf = clf.fit(x_train,y_train)
print(clf.predict(x_test))
print(y_test)