NVIDIA-Merlin / models

Merlin Models is a collection of deep learning recommender system model reference implementations
https://nvidia-merlin.github.io/models/main/index.html
Apache License 2.0
262 stars 50 forks source link

Example of dataloading pretrained embeddings in Merlin Models #788

Open viswa-nvidia opened 2 years ago

viswa-nvidia commented 2 years ago
bschifferer commented 1 year ago

@viswa-nvidia

We talked in the example sync meeting about the ticket:

Re: Add an example demonstrating support for pretrained embeddings for DLRM.

We will update this example ( https://github.com/NVIDIA-Merlin/models/blob/main/examples/usecases/entertainment-with-pretrained-embeddings.ipynb ) to use DLRM and then extend it to use dataloader pre-trained embeddings as well

Re: Add an example for usage of pre-trained embeddings with session-based recommendation (e.g. SIGIR Challenge dataset)

We will add an example for pre-trained embeddings for session-based models for SIGIR Challenge dataset. We will add serving with this example (https://github.com/NVIDIA-Merlin/Merlin/issues/886)