efficiency in _iter_train_pairs

Georgetown-IR-Lab / cedr

Code for CEDR: Contextualized Embeddings for Document Ranking, accepted at SIGIR 2019.

MIT License

156 stars 28 forks source link

Closed seanmacavaney closed 5 years ago

seanmacavaney commented 5 years ago

replacing random.shuffle with random.choice #4
using set for looking up if candidate negative sample appears as a positive sample