[Tensorflow/VAE-CF] Question about how to get the embeddings?

NVIDIA / DeepLearningExamples

State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.

13.54k stars 3.23k forks source link

[Tensorflow/VAE-CF] Question about how to get the embeddings? #477

Closed gilcanetasdias closed 3 years ago

gilcanetasdias commented 4 years ago

Hi there,

I'm trying to get the embeddings/latent representations of each user. Does anyone have recommendations?

nv-kkudrynski commented 3 years ago

Unfortunately VAE doesn't explicitly store embedding information. This task will be much easier with NCF or DLRM as they do use embeddings explicitly.