头歌机器学习---sklearn中的kNN算法

最新推荐文章于 2024-07-25 21:10:30 发布

liuyizeliuyize

最新推荐文章于 2024-07-25 21:10:30 发布

阅读量6.4k

点赞数 8

分类专栏：机器学习文章标签：机器学习 sklearn 人工智能

本文链接：https://blog.csdn.net/liiuyizeliuyize/article/details/129886103

版权

机器学习专栏收录该内容

8 篇文章 11 订阅

订阅专栏

第1关使用sklearn中的kNN算法进行分类

from sklearn.neighbors import KNeighborsClassifier

def classification(train_feature, train_label, test_feature):
    '''
    使用KNeighborsClassifier对test_feature进行分类
    :param train_feature: 训练集数据
    :param train_label: 训练集标签
    :param test_feature: 测试集数据
    :return: 测试集预测结果
    '''

    #********* Begin *********#
    clf = KNeighborsClassifier()
    clf.fit(train_feature, train_label)
    return clf.predict(test_feature)
    #********* End *********#

第2关使用sklearn中的kNN算法进行回归

from sklearn.neighbors import KNeighborsRegressor

def regression(train_feature, train_label, test_feature):
    '''
    使用KNeighborsRegressor对test_feature进行分类
    :param train_feature: 训练集数据
    :param train_label: 训练集标签
    :param test_feature: 测试集数据
    :return: 测试集预测结果
    '''

    #********* Begin *********#
    clf=KNeighborsRegressor() 
    clf.fit(train_feature, train_label)               
    return clf.predict(test_feature)
    #********* End *********#