1. 程式人生 > >sklearn系列學習--支援向量機SVM

sklearn系列學習--支援向量機SVM

#coding:utf-8
##1 匯入svm和資料集
from sklearn import svm,datasets

##2 呼叫SVC()
clf = svm.SVC()

##3 載入鳶尾花資料集
iris = datasets.load_iris()
X = iris.data
y = iris.target

##4 fit()訓練
clf.fit(X,y)

##5 predict()預測
pre_y = clf.predict(X[5:10])
print(pre_y)
print(y[5:10])

##6 匯入numpy
import numpy as np
test = np.array([[5.1,2.9,1.8,3.6]])

##7 對test進行預測
test_y = clf.predict(test)
print(test_y)

執行結果如下所示:

本部落格內容會繼續更新,歡迎交流