capreolus-ir / capreolus

A toolkit for end-to-end neural ad hoc retrieval
https://capreolus.ai
Apache License 2.0
95 stars 32 forks source link

Generate Ranked List of Train Data #188

Closed Pourbahman closed 3 years ago

Pourbahman commented 3 years ago

Hi Andrew,

I read your code line by line and according to my knowledge if I make include_train True in https://github.com/capreolus-ir/capreolus/blob/75ec83bf8fe8d2d012f4ad3b9ab98014bb191863/capreolus/task/rerank.py#L50 your code generates automatically ranked list for train data (like dev and test data) after model would be trained.

Would you please tell me if my understanding is true?

Thanks in advance, Kind Regards, Zahra

andrewyates commented 3 years ago

Hi Zahra, yes, that's correct. :)