See the website https://urikz.github.io/docent.
The website contains additional data and scripts for the EACL-2021 paper DOCENT: Learning Self-Supervised Entity Representations from Large Document Collections (arxiv)
Please, see https://goo.gle/research-docent for the Reviews2MovieLens dataset and modeling code.