Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add msmarco v2.1 trec rag #269

Draft
wants to merge 13 commits into
base: master
Choose a base branch
from

Conversation

mam10eks
Copy link
Contributor

@mam10eks mam10eks commented Aug 5, 2024

Dear all,

with a focus on the current TREC Rag task, this is an initial implementation for accessing documents via the corpus_iter and the document store for the document and the segmented document datasets of MS MARCO v2.1.

This is not completely finished (i.e., mainly documentation todos in the corresponding tickets are open), but as the deadline is soon, this might be useful for others even when the documentation is not yet completed.

@seanmacavaney
Copy link
Collaborator

Hey- can you try out the revision? I was getting errors running the tests before, so I refactored a bit. Now it's using the classes from v2 where possible.

@mam10eks
Copy link
Contributor Author

Perfect, nice that we could re-use more code from v2 classes, thanks!

The changes look good, I started the tests, I think they will run for a while, and will report back when they still work. (but I think everything should work)

@mam10eks
Copy link
Contributor Author

Hi, an update from the tests, they still run on my machine and also the other scripts that I had still work, so everything seems to be fine from my side.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants