KNN分类算法——鸢尾花数据示例

KNN算法——鸢尾花数据分类

1. 导包

import pandas as pd
from sklearn import datasets
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler
from sklearn.neighbors import KNeighborsClassifier

2. 原始数据

iris = datasets.load_iris()
X = iris.data
y = iris.target
print(X[0:10])
print(y[0:10])
print(iris.DESCR)

3. 数据预处理

# 拆分训练集、测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.25)
print(X_train[0:10])
print(y_train[0:10])

# 特征工程（标准化）
std = StandardScaler()
X_train = std.fit_transform(X_train)
X_test = std.transform(X_test)
print(X_train[0:10])

4. 构建KNN模型

knn = KNeighborsClassifier(n_neighbors=5)
knn.fit(X_train, y_train)

5. 结果预测与评分

# 预测
y_predict = knn.predict(X_test[0:10])
print("实际结果", y_test[0:10])
print("预测结果", y_predict)
# 评分
print("准确率", knn.score(X_test, y_test))