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)