
86
Chapitre 3. Classification
3.2 ENTRAÎNEMENT D’UN CLASSIFICATEUR BINAIRE
Nous allons simplier le problème pour l’instant en nous contentant d’identier un
chiffre, par exemple le 5. Ce «détecteur de 5» sera un exemple de classicateur binaire
capable d’effectuer la distinction entre deux classes uniquement, 5 et non-5. Créons
les deux vecteurs cible pour cette tâche de classication:
y_train_5 = (y_train == 5) # vrai pour les 5, faux pour le reste
y_test_5 = (y_test == 5)
Maintenant, choisissons un classicateur et entraînons-le. Une bonne idée
consiste à commencer par un classicateur de descente de gradient stochastique
(Stochastic Gradient ...