Skip to content

Commit

Permalink
Added new image
Browse files Browse the repository at this point in the history
  • Loading branch information
Pringled committed Oct 2, 2024
1 parent f51da09 commit d1a9539
Show file tree
Hide file tree
Showing 3 changed files with 2 additions and 1 deletion.
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -171,4 +171,5 @@ model2vec_models
results/*
counts/*
results_old/*
local/*
lightning_logs/*
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -278,7 +278,7 @@ As can be seen, Model2Vec models outperform the GloVe and WL256 models on all cl
The figure below shows the relationship between the number of sentences per second and the average classification score. The circle sizes correspond to the number of parameters in the models (larger = more parameters).
This plot shows that the Model2Vec models are much faster than the other models, while still being competitive in terms of classification performance with the [all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2) model.

| ![Description](assets/images/speed_vs_accuracy.png) |
| ![Description](assets/images/speed_vs_accuracy_v2.png) |
|:--:|
|*Figure: The average accuracy over all classification datasets plotted against sentence per second. The circle size indicates model size.*|

Expand Down
Binary file added assets/images/speed_vs_accuracy_v2.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit d1a9539

Please sign in to comment.