Modèles de classification

Validation de modèles en Python

Kasey Jones

Data Scientist

Modèles de classification

  • Réponses catégorielles :
    • Couleur de cheveux d'un nouveau-né
    • Gagnant d'un match de basketball
    • Genre de la prochaine chanson à la radio
Validation de modèles en Python

Le jeu de données Tic-Tac-Toe

... Bas-gauche Bas-centre Bas-droite Classe
... X O O positive
... O X O positive
... O O X positive
... X X O negative
... ... ... ... ...
Validation de modèles en Python

Google peut jouer au tic-tac-toe avec vous. Rendez-vous sur Google et cherchez Tic-Tac-Toe.

Validation de modèles en Python

Utiliser .predict() pour la classification

from sklearn.ensemble import RandomForestClassifier
rfc = RandomForestClassifier(random_state=1111)
rfc.fit(X_train, y_train)
rfc.predict(X_test)

array([1, 1, 1, 1, 0, 1, ...])

pd.Series(rfc.predict(X_test)).value_counts()

1    627
0    331
Validation de modèles en Python

Prédire des probabilités

rfc.predict_proba(X_test)

array([[0. , 1. ],
       [0.1, 0.9],
       [0.1, 0.9],
       ...])
Validation de modèles en Python
rfc = RandomForestClassifier(random_state=1111)
rfc.get_params()

{'bootstrap': True,
 'class_weight': None,
 'criterion': 'gini',
 ...}

rfc.fit(X_train, y_train)
rfc.score(X_test, y_test)

0.8989
Validation de modèles en Python

Classifions les scénarios de fin de partie au Tic-Tac-Toe

Validation de modèles en Python

Preparing Video For Download...