每天复习一个机器学习——KNN

from sklearn.neighbors import KNeighborsClassifier
from sklearn.datasets import load_iris
import matplotlib.pyplot as plt
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler

# 读取鸢尾花数据集
iris = load_iris()

iris_d = pd.DataFrame(iris['data'], columns = ['Sepal_Length', 'Sepal_Width', 'Petal_Length', 'Petal_Width'])
iris_d['Species'] = iris.target

# 实例化knn
knn = KNeighborsClassifier(n_neighbors=9)

# 数据集拆分
x_train, x_test, y_train, y_test = train_test_split(iris.data, iris.target, test_size=0.2, random_state=100)

# 标准化特征工程
transfer = StandardScaler()
x_train = transfer.fit_transform(x_train)
x_test = transfer.transform(x_test)

# 模型训练
knn.fit(x_train, y_train)

# 对比真实值和预测值
y_predicr = knn.predict(x_test)
print(y_predicr == y_test)


# 计算准确率
score = knn.score(x_test,y_test)
print(score)