>>> from sklearn.ensemble import RandomForestClassifier
>>> Y = [0, 1, 2]
>>> X = [[1, 0,0], [0, 1,0],[0, 0,1]]
>>> clf = RandomForestClassifier(n_estimators=10)
>>> clf = clf.fit(X, Y)
>>> clf.predict_proba([[1, 0,0],[0, 1,0],[0, 0,1]])
array([[ 0.9, 0.1, 0. ],
[ 0.1, 0.7, 0.2],
[ 0.1, 0.3, 0.6]])
随机森林最简实现
最新推荐文章于 2024-08-07 00:22:51 发布