rqvae-amazon-beauty / README.md
edobotta's picture
Update README.md
5f787bb verified
metadata
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 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 by Edoardo Botta, where more details on pre-training are found.