支持向量機(jī)
一、算法原理
二抖棘、scikit-learn SVM
from sklearn.svm import LinearSVC
svm_clf = LinearSVC(C=1)
svm_clf.fit(X, y)
三茂腥、鳶尾花示例
import numpy as np
from sklearn.pipeline import Pipeline
from sklearn.preprocessing import StandardScaler
from sklearn.svm import LinearSVC
# 導(dǎo)入數(shù)據(jù)集
from sklearn.datasets import fetch_openml
iris = fetch_openml(name='iris')
# 切分?jǐn)?shù)據(jù)集
X = iris['data'][:, 2:]
y = (iris['target'] == 'Iris-versicolor').astype(np.float64)
svm_clf = Pipeline([
('scalar', StandardScaler()),
('liner_svm', LinearSVC(C=1))
])
# 訓(xùn)練模型
svm_clf.fit(X, y)
# 預(yù)測(cè)
svm_clf.predict([[5.5, 1.7]])
運(yùn)行結(jié)果