Rank phrases and text based on query by leveraging hugging-face models.
Project description
spacy-rerank
Rank phrases and text based on query by leveraging hugging-face models. Currently, we are only leveraging tiny-bert model.
Installation and Implementation
- Install the package using the below
pip install spacyrerank
- Code for simple implementation
from spacyrerank.rerank import Reranker
query = "done effort is wasted"
texts = ["work done", "valuable man", "effort wasted", "Great work", "great work mate"]
reranker = Reranker(query, texts=texts)
reranker()
[{'rank': 2, 'text': 'effort wasted', 'similarity-score': 0.877},
{'rank': 0, 'text': 'work done', 'similarity-score': 0.86},
{'rank': 3, 'text': 'Great work', 'similarity-score': 0.773},
{'rank': 4, 'text': 'great work mate', 'similarity-score': 0.757},
{'rank': 1, 'text': 'valuable man', 'similarity-score': 0.719}]
CPU times: user 331 ms, sys: 76 ms, total: 407 ms
Wall time: 564 ms
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
spacyrerank-0.0.6.tar.gz
(3.6 kB
view details)
Built Distribution
File details
Details for the file spacyrerank-0.0.6.tar.gz
.
File metadata
- Download URL: spacyrerank-0.0.6.tar.gz
- Upload date:
- Size: 3.6 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.0 CPython/3.9.19
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1e5db9ead33143a492651da16c58640044cd0f956ac576775e73e1d84bc4ea49 |
|
MD5 | eddd0208efb135d216fd561356a2256b |
|
BLAKE2b-256 | bc2d1d5d998c5934fd5472ff000a6fb24fd515e09bdb34c38e00d06f30ea629f |
File details
Details for the file spacyrerank-0.0.6-py3-none-any.whl
.
File metadata
- Download URL: spacyrerank-0.0.6-py3-none-any.whl
- Upload date:
- Size: 3.9 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.0 CPython/3.9.19
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3f4ffe5b34062657072dc7d3d2bf14953f8c9d193e13b8fbf0b40daf84e92a4b |
|
MD5 | ab8d9e984ee49e23a0acd7de4a0fd686 |
|
BLAKE2b-256 | bb9c95910151e714402cd955bbedefd6583fcadd185c9d46551540199b8c8507 |