Skip to content

Instantly share code, notes, and snippets.

@fitomad
Created February 18, 2020 11:06
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save fitomad/ac247f6519a35e44f49725a65ec09fef to your computer and use it in GitHub Desktop.
Save fitomad/ac247f6519a35e44f49725a65ec09fef to your computer and use it in GitHub Desktop.
import turicreate as tc
# Cargamos los datos recogidos en la web
actions = tc.SFrame.read_csv('../Datasets/favorites-rating.csv')
# Creamos los juegos de datos para entrenamiento y validación
training_data, validation_data = tc.recommender.util.random_split_by_user(actions, 'user_id', 'show_id')
# El archivo favorites.csv tiene una columna extra llama ratings
# con valores inventados para este ejemplo
model = tc.ranking_factorization_recommender.create(training_data,
user_id='user_id',
item_id='show_id',
target='rating',
binary_target=True)
recommendations = model.get_similar_items()
# Vamos a suponer que hemos terminado de ver Stranger Things (66732) y
# queremos recomendar al usuario otros shows parecidos a este en base
# a su lista de preferencias
(recommendations[(recommendations['show_id'] == 66732)]).sort('rank', ascending=True).print_rows()
model.export_coreml("../Models/MyExplicitRecommender.mlmodel")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment