Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

recommend users for certain item #109

Closed
DeoLeung opened this issue Sep 22, 2016 · 3 comments
Closed

recommend users for certain item #109

DeoLeung opened this issue Sep 22, 2016 · 3 comments

Comments

@DeoLeung
Copy link

Hi, sorry for the stupid question:

Can this model predict top users for an item by design?
the predict & predict_rank method seem to be designed for user not item
if I do:

prediction = model.predict(all_user_ids, np.repeat(item_id, user_size))
# then sort the prediction

Is the top results reliable?

thank you

@paoloRais
Copy link
Contributor

I think you can obtain what you want by switching the role of the users and items, that is by giving to the model :

  • your users as the model's items input,
  • you items as the model's users input.

In this way you would get the best users for each item and can use the code as is.

@maciejkula
Copy link
Collaborator

Correct, this is the right approach. Think of your items as your users and you can use the model as-is.

@DeoLeung
Copy link
Author

Thank you for the reply, I will then just train two models~

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants