scikit-learn/KNN算法使用GridSearchCV调优

最新推荐文章于 2024-08-21 21:23:54 发布

szj_jojo

最新推荐文章于 2024-08-21 21:23:54 发布

阅读量8.9k

点赞数

分类专栏：深度学习机器学习文章标签： gridsearch sklearn mnist

本文链接：https://blog.csdn.net/szj_huhu/article/details/74909773

版权

环境：python 3 ， scikit-learn 0.18

sklearn调参有一个核武器—gridsearchcv，它存在的意义就是自动调参，只要把参数输进去，就可以对算法进行相应的调优，找到合适的参数。然而，在小数据集上还可以小打小闹。一旦数据的量级上去了，就很难得到结果，而且会报出memory error等错误

#coding:utf-8
"""
sklearn 0.18
python 3
KNN参数调优
"""
from sklearn.model_selection import GridSearchCV
from sklearn.model_selection import train_test_split
from sklearn.neighbors import KNeighborsClassifier
from sklearn.metrics import accuracy_score,confusion_matrix,classification_report