sklearn 随机森林代码示例

代码参考https://github.com/mbeyeler/opencv-machine-learning/blob/master/notebooks/10.03-Using-Random-Forests-for-Face-Recognition.ipynb

from sklearn.ensemble import RandomForestClassifier
from sklearn.ensemble import AdaBoostClassifier
from sklearn.datasets import load_breast_cancer
from sklearn.model_selection import train_test_split

cancer = load_breast_cancer()
X = cancer.data
y = cancer.target

X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=456)

ada = AdaBoostClassifier(n_estimators=100, random_state=456)
ada.fit(X_train, y_train)
print(ada.score(X_test, y_test))

forest = RandomForestClassifier(n_estimators=100, max_depth=1, random_state=456)
forest.fit(X_train, y_train)
print(forest.score(X_test, y_test))

forest = RandomForestClassifier(n_estimators=100, random_state=456)
forest.fit(X_train, y_train)
print(forest.score(X_test, y_test))

结果打印如下：

0.9790209790209791
0.9370629370629371
0.986013986013986