philipphager/baidu-ultr_uva-mlm-ctr

Query-document vectors and clicks for a subset of the Baidu Unbiased Learning to Rank dataset: https://arxiv.org/abs/2207.03051 This dataset uses a Jax-based BERT cross-encoder with 12 layers pre-trained for 2 million steps on the Baidu ULTR dataset to create query-document embeddings (768 dims). We link the model checkpoint also under `model/`.