Document Retriever model of ELI5-Category Dataset, need additional projection layer (see GitHub repo)