Skip to main content

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)

Uploaded Source

Built Distribution

spacyrerank-0.0.6-py3-none-any.whl (3.9 kB view details)

Uploaded Python 3

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

Hashes for spacyrerank-0.0.6.tar.gz
Algorithm Hash digest
SHA256 1e5db9ead33143a492651da16c58640044cd0f956ac576775e73e1d84bc4ea49
MD5 eddd0208efb135d216fd561356a2256b
BLAKE2b-256 bc2d1d5d998c5934fd5472ff000a6fb24fd515e09bdb34c38e00d06f30ea629f

See more details on using hashes here.

Provenance

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

Hashes for spacyrerank-0.0.6-py3-none-any.whl
Algorithm Hash digest
SHA256 3f4ffe5b34062657072dc7d3d2bf14953f8c9d193e13b8fbf0b40daf84e92a4b
MD5 ab8d9e984ee49e23a0acd7de4a0fd686
BLAKE2b-256 bb9c95910151e714402cd955bbedefd6583fcadd185c9d46551540199b8c8507

See more details on using hashes here.

Provenance

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page