鸢尾花识别代码
时间: 2023-09-28 09:11:58 浏览: 65
鸢尾花识别
以下是一个简单的鸢尾花识别代码,使用了scikit-learn库中的KNeighborsClassifier模型:
```
from sklearn.datasets import load_iris
from sklearn.neighbors import KNeighborsClassifier
# 加载鸢尾花数据集
iris = load_iris()
# 创建特征矩阵和目标向量
X = iris.data
y = iris.target
# 创建KNN分类器模型
knn = KNeighborsClassifier()
# 拟合模型并进行预测
knn.fit(X, y)
prediction = knn.predict([[5.1, 3.5, 1.4, 0.2]])
# 输出预测结果
print(iris.target_names[prediction])
```
在这个例子中,我们使用了鸢尾花数据集中的所有特征来训练一个KNN分类器模型,并使用该模型对新的鸢尾花样本进行预测。这个例子仅仅是一个简单的示例,实际中可能需要更多的特征和更复杂的模型来实现更准确的预测。
阅读全文