python 加入网格搜索与交叉验证的KNN算法

from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler
from sklearn.neighbors import KNeighborsClassifier
from sklearn.model_selection import GridSearchCV
def knn_iris_gscv():
    """KNN算法对鸢尾花进行分类
    加入网格搜索K值与测试集交叉验证"""
    #数据集获得
    iris = load_iris()#数据集 可改
    #数据集划分
    x_train,x_test,y_train,y_test = train_test_split(iris.data, iris.target, random_state= 6)
    #特征值标准化处理
    transfer = StandardScaler()
    x_train = transfer.fit_transform(x_train)
    x_test = transfer.transform(x_test)
    #KNN算法预估器
    estimator = KNeighborsClassifier()
    #加入网格搜索与交叉验证
    #参数准备
    param_dict = {"n_neighbors": [1,3,5,7,9,11,13]}#预测K值
    estimator = GridSearchCV(estimator, param_grid= param_dict ,cv= 10)
    estimator.fit(x_train,y_train)
    #评估模型
    #1.直接比对真实目标值与预估目标值
    y_predict = estimator.predict(x_test)
    print("预估值：\n",y_predict)
    print(y_predict == y_test) #真实目标值与预估值比对
    #2.计算准确率
    score = estimator.score(x_test,y_test)
    print("准确率：\n",score)

    #最佳参数
    print("最佳参数：\n",estimator.best_params_)
    # 最佳结果
    print("验证集中的最佳结果：\n", estimator.best_score_)
    # 最佳估计器
    print("估计器：\n", estimator.best_estimator_)
    # 交叉验证结果
    print("交叉验证结果：\n", estimator.cv_results_)
if __name__ == '__main__':
    # knn_iris()
    knn_iris_gscv()

结果如下：