BERT4Rec-VAE-Pytorch
BERT4Rec-VAE-Pytorch copied to clipboard
What configuration used to obtain the presented results for the VAE model?
Dear @jaywonchung,
For the movielens 1m dataset the reported result is NDCG@10 = 0.4049. I am currently unable to reproduce this result with the default configuration template train_vae_search_beta used and 'vae_num_hidden': 0. The results I get are closer to 0.37, so I would like to know how did you achieve those results.
What configuration have you used to achieve the above-mentioned results and how did you evaluate?
Thank you.