Algoritmo de recomendación de películas utilizando el filtrado colaborativo con la biblioteca Surprise

Vista previa del material en texto

Algoritmo de recomendación de películas utilizando el filtrado colaborativo con la biblioteca Surprise:
from surprise import Dataset, Reader, KNNBasic
from surprise.model_selection import train_test_split
# Cargar los datos en el formato esperado por Surprise
reader = Reader(rating_scale=(1, 5))
data = Dataset.load_from_df(df[['user_id', 'movie_id', 'rating']], reader)
# Dividir los datos en conjunto de entrenamiento y prueba
trainset, testset = train_test_split(data, test_size=0.2)
# Crear y entrenar el modelo KNN
algo = KNNBasic()
algo.fit(trainset)
# Realizar predicciones
predictions = algo.test(testset)