Python实现随机森林
时间: 2023-11-19 11:11:49 浏览: 82
随机森林是一种基于决策树的集成学习算法,可以用于分类和回归问题。Python中有许多库可以用于随机森林的实现,其中比较流行的是scikit-learn库。下面是一个简单的随机森林分类器的实现示例:
```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)
# 对新数据进行预测
print(clf.predict([[-0.2, 0.3, -0.4, 0.5]]))
```
在这个示例中,随机森林分类器使用了100个决策树,每个树的最大深度为2。我们使用`make_classification`函数生成了一些随机数据,然后使用`fit`方法训练模型,并使用`predict`方法对新数据进行预测。
阅读全文