用随机森林对鸢尾花进行分类
时间: 2023-05-23 14:03:30 浏览: 121
基于随机森林和bagging的鸢尾花分类(基于sklearn库)
可以使用以下代码使用随机森林对鸢尾花进行分类:
```python
from sklearn.datasets import load_iris
from sklearn.ensemble import RandomForestClassifier
iris = load_iris()
X = iris.data
y = iris.target
clf = RandomForestClassifier()
clf.fit(X, y)
# 预测
new_data = [[5.1, 3.5, 1.4, 0.2], [6.2, 2.9, 4.3, 1.3], [7.3, 2.9, 6.3, 1.8]]
pred = clf.predict(new_data)
print(pred) # 输出预测结果
```
以上代码使用 sklearn 库中的随机森林分类器对鸢尾花的数据进行分类。我们首先加载数据,然后使用随机森林分类器进行训练和预测。最后输出预测结果。
阅读全文