RandomForestClassifier
随机森林是一种元估计器,它适用于数据集的各个子样本上的多个决策树分类器,并使用平均来提高预测精度和控制过拟合。
一个简单的用法示例:
导入:
from sklearn.ensemble import RandomForestClassifier
定义列车数据和目标数据:
train = [[1,2,3],[2,5,1],[2,1,7]]
target = [0,1,0]
target
中的值表示你要预测的标签。
启动 RandomForest 对象并执行 learn(fit)
:
rf = RandomForestClassifier(n_estimators=100)
rf.fit(train, target)
预测:
test = [2,2,3]
predicted = rf.predict(test)