Skip to main content

Lightweight keyword & keyphrase extraction and ranking using TF-IDF with optional embedding re-ranking.

Project description

auto-keyword-ranker

PyPI version License

Lightweight Python package to extract and rank the most relevant keywords and keyphrases from text.

Goal: One-line call to get ranked keywords for articles, blog posts, or short documents.
Core approach uses TF-IDF; optional re-ranking with sentence-transformer embeddings.


Installation

pip install auto-keyword-ranker

With optional embedding-based re-ranking:

pip install auto-keyword-ranker[embed]

Quickstart

from autokeyword import rank_keywords

text = """
Artificial intelligence is transforming industries by enabling new capabilities
such as natural language processing, computer vision, and advanced data analytics.
"""

# Simple TF-IDF keyword ranking
keywords = rank_keywords(text, top_n=5)
print(keywords)

Output

A list of (keyword, score) pairs, for example:

[('artificial intelligence', 0.42),
 ('data analytics', 0.33),
 ('natural language processing', 0.29),
 ('computer vision', 0.25),
 ('industries', 0.21)]

API

rank_keywords(texts, top_n=10, method='tfidf', ngram_range=(1,2), stop_words=True, use_embeddings=False, embedding_model=None, combine_score_alpha=0.6)

See docstrings in autokeyword/core.py for full parameter descriptions.


CLI

You can also run the CLI (after installation):

python -m autokeyword.cli --text "Your article text here" --top 10

How It Works TF-IDF scoring formula:

$$ \text{TF-IDF}(t, d) = \text{TF}(t, d) \times \log \frac{N}{1 + \text{DF}(t)} $$

Where:

  • ( t ) = term
  • ( d ) = document
  • ( N ) = total number of documents

License

MIT License © 2025 Reya Oberoi

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

auto_keyword_ranker-0.1.7.tar.gz (5.5 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

auto_keyword_ranker-0.1.7-py3-none-any.whl (6.2 kB view details)

Uploaded Python 3

File details

Details for the file auto_keyword_ranker-0.1.7.tar.gz.

File metadata

  • Download URL: auto_keyword_ranker-0.1.7.tar.gz
  • Upload date:
  • Size: 5.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.9

File hashes

Hashes for auto_keyword_ranker-0.1.7.tar.gz
Algorithm Hash digest
SHA256 91cd608647588455c21ba99b871a4e51f6042a5ca7f2e22822511dec0ad788bf
MD5 94c41f4bfa9626698b56308743fee8bc
BLAKE2b-256 13d10bc9375c120887224cc73c3994aabd05d3e84d4a5d8d803a163cfc630546

See more details on using hashes here.

File details

Details for the file auto_keyword_ranker-0.1.7-py3-none-any.whl.

File metadata

File hashes

Hashes for auto_keyword_ranker-0.1.7-py3-none-any.whl
Algorithm Hash digest
SHA256 589059222589da10ea14a11fc234076eb310e6a1e022c594cb54c9981d8c2f06
MD5 8cc7dbf1fb4a26896206dffc2f86db34
BLAKE2b-256 339e02d9f18a6846131cba0de8eb949818eb8be0bc6d61d55534c312850e6c86

See more details on using hashes here.

Supported by

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