models icon indicating copy to clipboard operation
models copied to clipboard

[Task] Example notebooks for generation top_k recommendations with ranking and retrieval models

Open rnyak opened this issue 3 years ago • 2 comments

Description

Currently none of our example notebooks do not have any code script showcasing how one can do model prediction locally, generate top-K recommendations from the prediction results. This does not include Merlin systems, it is doing offline predictions.

We can either add a new notebook or extend the existing retrieval and ranking notebooks to show such functionality. I think extending the existing notebooks would be better idea.

We have been getting requests for such examples internally or externally and this will be very informative for our users.

rnyak avatar Aug 25 '22 11:08 rnyak

This is still being worked on via https://github.com/NVIDIA-Merlin/models/issues/718.

I will create an example for generating topk recommendations from a ranking model, but for retrieval model we still need the https://github.com/NVIDIA-Merlin/models/issues/718 task to be completed to create an example.

Related PRs:

  • https://github.com/NVIDIA-Merlin/models/pull/705
  • https://github.com/NVIDIA-Merlin/models/pull/715

@sararb is currently working on TopK encoder.

rnyak avatar Sep 28 '22 15:09 rnyak

@sararb do you know if the API update is finished and we can implement this example?

bschifferer avatar Oct 18 '22 09:10 bschifferer

I am closing this since we have an example for generating topK recommendations, merged by https://github.com/NVIDIA-Merlin/models/pull/900

rnyak avatar Dec 13 '22 19:12 rnyak