Menu
CoddyTech

Метод fit

Урок 7 из 19 курса Введение в машинное обучение на Coddy.

У KNN нет явной фазы обучения. Он просто сохраняет набор данных. Только во время фазы predict он вычисляет расстояние от новой точки до всех остальных.

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