← retour aux snippets

sklearn: SVM classification

Classifieur SVM (RBF) avec scaling.

python ml #sklearn#svm#classification

objectif

Classifieur SVM (RBF) avec scaling.

code minimal

from sklearn.svm import SVC
from sklearn.pipeline import make_pipeline
from sklearn.preprocessing import StandardScaler
from sklearn.datasets import load_iris
X, y = load_iris(return_X_y=True)
clf = make_pipeline(StandardScaler(), SVC(probability=True)).fit(X, y)
print(hasattr(clf, "predict_proba"))

utilisation

print(clf.predict(X[:3]).tolist())

variante(s) utile(s)

from sklearn.svm import LinearSVC
print(hasattr(LinearSVC(), "fit"))

notes

  • Sensible à l’échelle; utiliser StandardScaler.