edobotta commited on
Commit
5f787bb
·
verified ·
1 Parent(s): 6dd831b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -3
README.md CHANGED
@@ -2,8 +2,15 @@
2
  tags:
3
  - model_hub_mixin
4
  - pytorch_model_hub_mixin
 
 
 
 
 
 
5
  ---
 
6
 
7
- This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
8
- - Library: [More Information Needed]
9
- - Docs: [More Information Needed]
 
2
  tags:
3
  - model_hub_mixin
4
  - pytorch_model_hub_mixin
5
+ - rq-vae
6
+ - recommender_systems
7
+ - generative_retrieval
8
+ - tiger
9
+ - retrieval
10
+ - variational_autoencoder
11
  ---
12
+ Trained RQ-VAE checkpoint from [Recommender Systems with Generative Retrieval](https://arxiv.org/pdf/2305.05065) by Google DeepMind.
13
 
14
+ 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.
15
+
16
+ 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.