Update README.md

dawenl · Feb 15, 2018 · 22082a9 · 22082a9
1 parent cd89d97
commit 22082a9
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -1,2 +1,5 @@
-# vae_cf
-Variational autoencoders for collaborative filtering
+# Variational autoencoders for collaborative filtering
+
+This notebook accompanies the paper "Variational autoencoders for collaborative filtering" by Dawen Liang, Rahul G. Krishnan, Matthew D. Hoffman, and Tony Jebara, in The Web Conference (aka WWW) 2018.
+
+In this notebook, we show a complete self-contained example of training a variational autoencoder (as well as a denoising autoencoder) with multinomial likelihood (described in the paper) on the public Movielens-20M dataset, including both data preprocessing and model training.