File size: 675 Bytes
945cbb6 5f787bb 945cbb6 5f787bb 945cbb6 5f787bb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
---
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.
|