用代码实现定义随机森林分类器并拟合训练数据
时间: 2023-07-29 18:10:21 浏览: 87
Python随机森林分类器代码实现
5星 · 资源好评率100%
好的,以下是一个使用sklearn库实现随机森林分类器并拟合训练数据的示例代码:
```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(n_estimators=100, max_depth=2, random_state=0)
# 拟合训练数据
clf.fit(X, y)
```
这段代码首先使用`make_classification()`函数创建了一个随机数据集,然后定义了一个随机森林分类器模型`clf`,并指定了模型的一些参数,例如`n_estimators`表示树的数量,`max_depth`表示每个决策树的最大深度等等。最后,使用`fit()`方法拟合训练数据。
阅读全文