Scikit Learn RandomForestClassifier 入门实例

from sklearn.ensemble import RandomForestClassifier
data=[[0,0,0],[1,1,1],[2,2,2],[1,1,1],[2,2,2],[3,3,3],[1,1,1],[4,4,4]]
target=[0,1,2,1,2,3,1,4]
rf = RandomForestClassifier()

rf.fit(data,target)

print rf.predict_proba([[1,1,1]])
#[[ 0.  1.  0.  0.  0.]]

输出有5个数因为target有一共有5个值