写一份随机森林的python代码
时间: 2023-12-15 16:56:41 浏览: 56
随机森林的python代码
好的,以下是一个简单的随机森林的Python代码:
```
from sklearn.ensemble import RandomForestClassifier
from sklearn.datasets import make_classification
X, y = make_classification(n_samples=1000, n_features=4,
n_informative=2, n_redundant=0,
random_state=0, shuffle=False)
clf = RandomForestClassifier(max_depth=2, random_state=0)
clf.fit(X, y)
print(clf.predict([[0, 0, 0, 0]]))
```
这个代码使用了sklearn库中的随机森林分类器,并使用make_classification函数生成了一组随机的分类数据。代码创建了一个随机森林分类器,并使用fit函数进行训练。最后,代码预测了一个新的数据点的类别并将其打印出来。
阅读全文