método fit
Lección 7 de 19 del curso Introducción al aprendizaje automático de Coddy.
El KNN no tiene una fase de entrenamiento explícita. Simplemente almacena el conjunto de datos. Solo durante la fase de predicción calcula la distancia entre el nuevo punto y el resto.
Desafío
IntermedioImplementa el cálculo de la distancia euclidiana en el método distance, y modifica el método fit para almacenar la lista de puntos en self.points y las etiquetas correspondientes en self.labels. X_train son los puntos y y_train son las etiquetas.
Pruébalo tú mismo
import numpy
class KNN:
def __init__(self, k):
self.points = None
self.labels = None
self.k = k
def distance(self, point_a, point_b):
# Escribe tu código aquí
def fit(self, X_train, y_train):
# Escribe tu código aquí
pass
def predict(self, X_test):
passTodas las lecciones de Introducción al aprendizaje automático
Practica por tu cuenta: Compilador de Python online