search

terrierteam / pyterrier_ance

5 stars 9 forks source link

added ANCE re-ranker #4

Closed seanmacavaney closed 3 years ago

seanmacavaney commented 3 years ago

closes #3

seanmacavaney commented 3 years ago

Hold on there! I wasn't ready for a review yet. Pushed to test on Colab.

seanmacavaney commented 3 years ago

@cmacdonald alright- ready now. May want to squash it upon merge because there was a bit of debugging.

groupby and pt.apply.by_query were not ideal because the ance package does some logging internally, which spams the output when done by individual query. (Also allows batching over documents across queries.)