KNN算法-搜索最优超参数:n_neighbors /weight/p

在本文中,将选取正确率作为衡量标准,自行实现搜索最优超参数,而非直接调用sklearn中的GridSearchCV。

先简单介绍KNN中的三个超参数:

超参数为:n_neighbors /weight/p(只有当weight=distance的时候,p值才有意义)

  1. n_neighbors:取邻近点的个数k。k取1-9测试
  2. weight:距离的权重;uniform:一致的权重;distance:距离的倒数作为权重
  3. p:闵可斯基距离的p值; p=1:即欧式距离;p=2:即曼哈顿距离;p取1-6测试
#超参数为:n_neighbors /weight/p
# n_neighbors:取邻近点的个数k。k取1-9测试
#weight:距离的权重;uniform:一致的权重;distance:距离的倒数作为权重
#p:闵可斯基距离的p值; p=1:即欧式距离;p=2:即曼哈顿距离;p取1-6测试
#只有当weight=distance的时候,p值才有意义

def searchBestPar():
    bestScore=0
    bestK=-1
    bestWeight=""

    # weight==uniform时
    for k in range(1,10):
        clf = KNeighborsClassifier(n_neighbors=k,weights="uniform")
        clf.fit(trainX,trainY)
        scor=clf.score(testX,testY)
        if scor > bestScore:
            bestScore=scor
            bestK=k
            bestWeight="uniform"

    # weight==distance时
    for k in range(1,10):
        for p in range(1,7):
            clf=KNeighborsClassifier(n_neighbors=k,weights="distance",p=p)
            clf.fit(trainX,trainY)
            scor = clf.score(testX, testY)
            if scor > bestScore:
                bestScore = scor
                bestK = k
                bestWeight = "distance"

    print("the best n_neighbors", bestK)
    print("the best weights", bestWeight)
    print("the best p", p)

if __name__ == '__main__':
    iris=datasets.load_iris()
    trainX, testX, trainY, testY = train_test_split(iris.data,iris.target)
    searchBestPar()
 

 

 

 

 

 

评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值