生成一段用knn算法分析鸢尾花数据
时间: 2023-11-17 11:05:32 浏览: 74
KNN算法在鸢尾花数据集的实现
5星 · 资源好评率100%
KNN算法是一种基于实例的学习方法,它可以用于对鸢尾花数据进行分类分析。下面我们来看一段使用KNN算法对鸢尾花数据进行分类的代码:
```python
# 导入数据集
from sklearn.datasets import load_iris
iris = load_iris()
# 分割数据集
from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(iris['data'], iris['target'], random_state=0)
# 标准化数据
from sklearn.preprocessing import StandardScaler
scaler = StandardScaler().fit(X_train)
X_train = scaler.transform(X_train)
X_test = scaler.transform(X_test)
# 训练KNN模型
from sklearn.neighbors import KNeighborsClassifier
knn = KNeighborsClassifier(n_neighbors=3)
knn.fit(X_train, y_train)
# 预测测试集
y_pred = knn.predict(X_test)
# 分析准确率
from sklearn.metrics import accuracy_score
accuracy = accuracy_score(y_test, y_pred)
print('Accuracy:', accuracy)
```
在这段代码中,我们首先导入了鸢尾花数据集,并将其分割为训练集和测试集。然后,我们对数据进行标准化处理,以便更好地训练模型。接下来,我们使用KNeighborsClassifier类来训练KNN模型,并设置n_neighbors参数为3。最后,我们对测试集进行预测,并计算准确率。
通过运行这段代码,我们可以得到KNN算法在鸢尾花数据集上的分类准确率。在实际应用中,我们可以根据这个准确率来评估模型的表现,并选择最优的K值来进一步优化模型的性能。
阅读全文