RandomForestClassifier

Ансамбль деревьев решений для классификации.

СигнатураRandomForestClassifier(n_estimators=100, max_depth=None, random_state=None)

Голосование множества деревьев, обученных на случайных подвыборках. Один из самых надёжных «коробочных» классификаторов. Атрибут feature_importances_ показывает значимость признаков.

from sklearn.ensemble import RandomForestClassifier

clf = RandomForestClassifier(n_estimators=300, random_state=42)
clf.fit(X_train, y_train)
print(clf.score(X_test, y_test))

← Все записи: scikit-learn