from sklearn.neighbors import KNeighborsClassifier knn_model = KNeighborsClassifier() knn_model.fit(X_train_std,y_train) print(knn_model.score(X_train_std,y_train)) print(knn_model.score(X_test_std,y_test)) from sklearn.metrics import classification_report, confusion_matrix y_pred =knn_model.predict(X_test) print(classification_report(y_test,y_pred)) cm = confusion_matrix(y_test, y_pred) plt.figure(figsize = (8,8)) sns.heatmap(cm,cmap= "Blues", linecolor = 'black' , linewidth = 1 , annot = True, fmt='' , xticklabels = ['A','B','C','D'] , yticklabels = ['A','B','C','D']) plt.xlabel("Predicted") plt.ylabel("Actual") plt.show() from sklearn.neighbors import KNeighborsClassifier knn_model = KNeighborsClassifier() prams ={ 'n_neighbors':[13,15,18,22]} gd_knn= GridSearchCV(knn_model , param_grid=prams, n_jobs=-1 ,cv=10) gd_knn.fit(X_train_std , y_train) print(gd_knn.best_estimator_) tuned_knn = gd_knn.best_estimator_ print(f'Train : {tuned_knn.score(X_train_std, y_train)}') model_eval(tuned_knn,X_test_std,y_test)
时间: 2024-04-02 22:35:54 浏览: 66
kNN.zip_K._KNN 分类_knn_python欧氏距离_欧氏距离
这段代码使用了scikit-learn库中的KNeighborsClassifier类来创建一个K近邻分类器,并使用fit方法对标准化后的训练数据进行拟合。接着,使用score方法计算了模型在训练集和测试集上的准确率,并将其打印输出。之后,使用classification_report和confusion_matrix方法计算并打印了模型的分类报告和混淆矩阵。
随后,使用GridSearchCV方法对K近邻分类器进行调参,尝试不同的n_neighbors参数组合,以求得最优的模型。最后,使用tuned_knn.score方法计算了调参后模型在训练集上的准确率,并使用model_eval方法计算并打印了模型在测试集上的性能表现。
阅读全文