Skip to content

dataset: Add mbpp retrieval #3037

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.

Sign up for GitHub

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

Jump to bottom
Merged
merged 6 commits into from
Aug 20, 2025
Merged

dataset: Add mbpp retrieval #3037

merged 6 commits into from
Aug 20, 2025

Conversation

fzoll
Copy link
Contributor

@fzoll fzoll commented Aug 17, 2025 •
edited by KennethEnevoldsen
Loading

fixes #3015

  • I have run the following models on the task (adding the results to the pr). These can be run using the mteb run -m {model_name} -t {task_name} command.
    • sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2
    • intfloat/multilingual-e5-small
  • I have checked that the performance is neither trivial (both models gain close to perfect scores) nor random (both models gain close to random scores).
  • I have considered the size of the dataset and reduced it if it is too big (2048 examples is typically large enough for most tasks)

fzoll and others added 5 commits August 17, 2025 17:46
@fzoll
Add MBPP retrieval task
827b9fb 
- Code retrieval task based on 378 Python programming problems
- Natural language queries matched to Python code implementations
- Uses python-Code evaluation language for code-specific metrics
- Includes proper citations and descriptive statistics
@fzoll
Add MBPPRetrieval to imports
adf2f97
@fzoll
Add descriptive statistics for MBPPRetrieval
724102a
@fzoll
Merge branch 'main' into add-mbpp-retrieval
f6fcf97
@fzoll
Reformatting
5dc3e4e
@KennethEnevoldsen KennethEnevoldsen changed the title Add mbpp retrieval dataset: Add mbpp retrieval Aug 17, 2025
KennethEnevoldsen
KennethEnevoldsen approved these changes Aug 17, 2025
View reviewed changes
Copy link
Contributor

@KennethEnevoldsen KennethEnevoldsen left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looking great - these PRs are a pleasure to review :)

(again citations only citations formatting is missing)

@fzoll
Copy link
Contributor Author

fzoll commented Aug 17, 2025

@KennethEnevoldsen Ah, sorry, i forgot about the citation formatting. Can you please check now?

isaac-chung
isaac-chung approved these changes Aug 20, 2025
View reviewed changes
Copy link
Collaborator

@isaac-chung isaac-chung left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Citation looks good as well (CI passes). Thanks!

@isaac-chung isaac-chung merged commit ac69263 into embeddings-benchmark:main Aug 20, 2025
9 checks passed
isaac-chung
isaac-chung reviewed Aug 20, 2025
View reviewed changes
mteb/tasks/Retrieval/code/MBPPRetrieval.py
reference="https://huggingface.co/datasets/embedding-benchmark/MBPP",
dataset={
"path": "embedding-benchmark/MBPP",
"revision": "12a7c6b3b7e985be9b9f81e2f06306ffd17cc5a4",
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hmm this version does not seem to exist. I was able to find the latest one though:

586a1fd6a0c63fdeda3b49c0293559a81c79cdec

@isaac-chung isaac-chung mentioned this pull request Aug 20, 2025
Merged
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@isaac-chung isaac-chung isaac-chung approved these changes

@KennethEnevoldsen KennethEnevoldsen Awaiting requested review from KennethEnevoldsen

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

Add dataset: MBPP
3 participants
@fzoll @KennethEnevoldsen @isaac-chung