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

Added utility function for downloading and loading models results #995

Merged
merged 2 commits into from
Jun 28, 2024

Conversation

KennethEnevoldsen
Copy link
Contributor

@KennethEnevoldsen KennethEnevoldsen commented Jun 27, 2024

Model results can now be loaded using

mteb.load_results()

This will download the results repo (on github) -- or update it if needed -- and then load then in as a dict (see docstring).

The hope is that this is the start to developing better result fetching + result downloading from MTEB. I could also imagine we could add the option to load specific model result from HF hub, push results to models card (creating a PR) etc.

Checklist

  • Run tests locally to make sure nothing is broken using make test.
  • Run the formatter to format the code using make lint.

Copy link
Contributor

@orionw orionw left a comment

Choose a reason for hiding this comment

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

I like this, thanks @KennethEnevoldsen. Then we can test that new changes can still load and also run some Github Actions using it.

@KennethEnevoldsen KennethEnevoldsen enabled auto-merge (squash) June 28, 2024 08:23
@KennethEnevoldsen KennethEnevoldsen merged commit abef8dc into main Jun 28, 2024
7 checks passed
@KennethEnevoldsen KennethEnevoldsen deleted the results_utils branch June 28, 2024 08:29
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.

None yet

2 participants