分類モデル

Python によるモデル検証

Kasey Jones

Data Scientist

分類モデル

  • カテゴリカル応答:
    • 新生児の髪色
    • バスケットボール試合の勝者
    • ラジオの次の曲のジャンル
Python によるモデル検証

三目並べデータセット

... 左下 下中央 右下 クラス
... X O O positive
... O X O positive
... O O X positive
... X X O negative
... ... ... ... ...
Python によるモデル検証

Googleは三目並べで対戦できます。Googleで「Tic-Tac-Toe」を検索してください。

Python によるモデル検証

.predict()で分類

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
Python によるモデル検証

確率を予測する

rfc.predict_proba(X_test)

array([[0. , 1. ],
       [0.1, 0.9],
       [0.1, 0.9],
       ...])
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
Python によるモデル検証

三目並べの終局を分類してみましょう

Python によるモデル検証

Preparing Video For Download...