Menu
CoddyTech

fit 方法

Coddy 机器学习入门 课程的第 7 课,共 19 课。

KNN 没有明确的训练阶段。它只是存储数据集。只有在预测阶段,它才会计算新点与其余点之间的距离。

challenge icon

挑战

中等

在 distance 方法中实现欧几里得距离的计算,并修改 fit 方法,将点列表存储在 self.points 中,并将对应的标签存储在 self.labels 中。X_train 是点,y_train 是标签。

亲手试试

import numpy
class KNN:
    def __init__(self, k):
        self.points = None
        self.labels = None
        self.k = k
    
    def distance(self, point_a, point_b):
        # 在这里编写你的代码
    
    def fit(self, X_train, y_train):
        # 在这里编写你的代码
        pass

    def predict(self, X_test):
        pass

机器学习入门中的所有课

自己练习: Python 在线编译器