← retour aux snippets

sklearn: KNN classifier basique

Classifier k plus proches voisins avec pipeline.

python ml #sklearn#knn#classification

objectif

Classifier k plus proches voisins avec pipeline.

code minimal

from sklearn.neighbors import KNeighborsClassifier
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(), KNeighborsClassifier(n_neighbors=5)).fit(X, y)
print(hasattr(clf, "predict"))

utilisation

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

variante(s) utile(s)

from sklearn.neighbors import RadiusNeighborsClassifier
print(hasattr(RadiusNeighborsClassifier(), "fit"))

notes

  • Choisir k impair; scaler nécessaire.