Menu
CoddyTech

Méthode fit

Leçon 7 sur 19 du cours Introduction à l’apprentissage automatique de Coddy.

Le KNN n'a pas de phase d'entraînement explicite. Il stocke simplement le jeu de données. Ce n'est que lors de la phase predict qu'il calcule la distance entre le nouveau point et les autres.

challenge icon

Défi

Moyen

Implémentez le calcul de la distance euclidienne dans la méthode distance, et modifiez la méthode fit pour stocker la liste des points dans self.points et les étiquettes correspondantes dans self.labels. X_train sont les points et y_train sont les étiquettes.

Essayez vous-même

import numpy
class KNN:
    def __init__(self, k):
        self.points = None
        self.labels = None
        self.k = k
    
    def distance(self, point_a, point_b):
        # Écrivez votre code ici
    
    def fit(self, X_train, y_train):
        # Écrivez votre code ici
        pass

    def predict(self, X_test):
        pass

Toutes les leçons de Introduction à l’apprentissage automatique

Entraînez-vous par vous-même : Compilateur Python en ligne