Descarga la aplicación para disfrutar aún más
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)
Compartir