KNN算法--python实现代码

最新推荐文章于 2022-01-27 12:14:54 发布

原创

最新推荐文章于 2022-01-27 12:14:54 发布 · 968 阅读

2 ·

CC 4.0 BY-SA版权

本文提供了一个简单的KNN算法实现，通过计算测试数据与训练集中的距离，选择最近的K个样本进行投票，以确定预测结果。代码示例中包括了鸢尾花数据集的应用，并展示了如何使用该算法进行预测。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >


一个简单的KNN 算法实现，主要从KNN的核心思想对测试数据计算距离，在训练集中进行搜索后，选取距离最近的K个label，做vote。即可得出结果。
(代码随时会更新，有问题欢迎讨论，QQ:602394876，谢谢！)

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from sklearn.datasets import load_iris
import math

class KNN():

    def __init__(self):
        pass

    def distance(self,train,test):
        """

        :param test:
        :param train: