|
|
--- |
|
|
tags: |
|
|
- model_hub_mixin |
|
|
- pytorch_model_hub_mixin |
|
|
- rq-vae |
|
|
- recommender_systems |
|
|
- generative_retrieval |
|
|
- tiger |
|
|
- retrieval |
|
|
- variational_autoencoder |
|
|
--- |
|
|
Trained RQ-VAE checkpoint from [Recommender Systems with Generative Retrieval](https://arxiv.org/pdf/2305.05065) by Google DeepMind. |
|
|
|
|
|
Given item textual embeddings, this model generates a semantic id for the item that can be used for further downstream recommendation, ranking and retrieval tasks. |
|
|
|
|
|
The model is trained on Amazon Beauty using the open-source [paper implementation](https://github.com/EdoardoBotta/RQ-VAE-Recommender?tab=readme-ov-file) by Edoardo Botta, where more details on pre-training are found. |
|
|
|
