Skip to content

dataset: Added wikisql retrieval #3039

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: Added wikisql retrieval #3039

merged 6 commits into from
Aug 20, 2025

Conversation

fzoll
Copy link
Contributor

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

fixes #3017

  • 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 added 4 commits August 17, 2025 17:49
@fzoll
Add WikiSQL retrieval task
de7f9d0 
- Code retrieval task based on WikiSQL natural language to SQL dataset
- Natural language questions matched to SQL query implementations
- Uses sql-Code evaluation language for SQL-specific metrics
- Includes proper citations and descriptive statistics
@fzoll
Add WikiSQLRetrieval to imports
571ef3b
@fzoll
Add descriptive statistics for WikiSQLRetrieval
b6d6604
@fzoll
Reformatting
e184c2e
KennethEnevoldsen
KennethEnevoldsen reviewed Aug 17, 2025
View reviewed changes
mteb/tasks/Retrieval/code/WikiSQLRetrieval.py
if self.data_loaded:
return

from datasets import load_dataset
Copy link
Contributor

Choose a reason for hiding this comment

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

Any reason why this is not a top-level import?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

@KennethEnevoldsen No specific reason, i can add this import to top level

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.

A few minor comments - otherwise this looks good.

@KennethEnevoldsen KennethEnevoldsen changed the title Add wikisql retrieval dataset: Added wikisql retrieval Aug 17, 2025
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.

Looks good, thanks!

@isaac-chung isaac-chung merged commit 7b289f5 into embeddings-benchmark:main Aug 20, 2025
9 checks passed
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: WikiSQL
3 participants
@fzoll @KennethEnevoldsen @isaac-chung