Dataset for HW6: Supervised BERT-based Retrieval Model