生成加权随机森林的python代码
时间: 2023-11-27 15:05:14 浏览: 52
随机森林的python代码
下面是生成基于Scikit-learn库的加权随机森林的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)
# 定义权重
sample_weight = [0.1 if i == 0 else 1 for i in y]
# 定义加权随即森林模型
clf = RandomForestClassifier(n_estimators=100, random_state=0)
# 训练模型
clf.fit(X, y, sample_weight=sample_weight)
# 预测结果
clf.predict([[0, 0, 0, 0]])
```
阅读全文