Моделі класифікації

Валідація моделей у Python

Kasey Jones

Data Scientist

Категоріальні відповіді:
- Колір волосся новонародженого
- Переможець баскетбольного матчу
- Жанр наступної пісні на радіо

Набір даних Tic-Tac-Toe

...	Нижньо-ліве	Нижньо-середнє	Нижньо-праве	Клас
...	X	O	O	positive
...	O	X	O	positive
...	O	O	X	positive
...	X	X	O	negative
...	...	...	...	...

Google зіграє з вами в Tic-Tac-Toe, якщо забажаєте. Перейдіть у Google і знайдіть Tic-Tac-Toe games.

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

rfc.predict_proba(X_test)

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

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