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 hashes)
Built Distribution
Close
Hashes for spacyrerank-0.0.6-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3f4ffe5b34062657072dc7d3d2bf14953f8c9d193e13b8fbf0b40daf84e92a4b |
|
MD5 | ab8d9e984ee49e23a0acd7de4a0fd686 |
|
BLAKE2b-256 | bb9c95910151e714402cd955bbedefd6583fcadd185c9d46551540199b8c8507 |