Skip to main content
simpleT5

Quickly extract key-phrases/topics from you text data with T5 transformer

KeyPhraseTransformer is built on T5 Transformer architecture, trained on 500,000 training samples to extract important phrases/topics/themes from text of any length.

Why KeyPhraseTransformer?

  • You get the power of amazing T5 architecture. 
  • The underlying T5 model is specifically trained in extracting important phrases from the text corpus, so the results are of superior quality.
  • No pre-processing is needed of any kind. Just dump your data to the model
  • It does not need any n-gram-related inputs from user. It can automatically extract unigram, bigram, or trigram on its own.
  • It can process text data of any length as it breaks down input text into smaller chunks internally
  • It helps to automate the topic modeling/keyword extraction process end to end with no manual intervention.

Installation:

pip install keyphrasetransformer

Use:

Generic badge

from keyphrasetransformer import KeyPhraseTransformer

kp = KeyPhraseTransformer()

doc = """
Transfer learning, where a model is first pre-trained on a data-rich task before being fine-tuned 
on a downstream task, has emerged as a powerful technique in natural language processing (NLP). 
The effectiveness of transfer learning has given rise to a diversity of approaches, methodology, and practice. 
In this paper, we explore the landscape of transfer learning techniques for NLP by introducing a unified framework 
that converts every language problem into a text-to-text format. Our systematic study compares pretraining objectives, 
architectures, unlabeled datasets, transfer approaches, and other factors on dozens of language understanding tasks. 
By combining the insights from our exploration with scale and our new “Colossal Clean Crawled Corpus”, 
we achieve state-of-the-art results on many benchmarks covering summarization, question answering, 
text classification, and more. To facilitate future work on transfer learning for NLP, 
we release our dataset, pre-trained models, and code.

"""

kp.get_key_phrases(doc)
['transfer learning',
 'natural language processing (nlp)',
 'nlp',
 'text-to-text',
 'language understanding',
 'transfer approach',
 'pretraining objectives',
 'corpus',
 'summarization',
 'question answering']

Release files for keyphrasetransformer 0.0.2

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for keyphrasetransformer 0.0.2
File Size Uploaded
keyphrasetransformer-0.0.2.tar.gz 3.9 kB Details

Release files / keyphrasetransformer-0.0.2.tar.gz

Download URL keyphrasetransformer-0.0.2.tar.gz
Size 3.9 kB
Tags Source
SHA-256 checksum
How to use checksums
1afae952c4a63bc97e3072fb75bb09e4be9703c05c1cde34740021ee57bad3ab
BLAKE2b-256 checksum
How to use checksums
f53256a0b4ccb814095f9e40bf8fdfa5d06f322fe9c6d8a693e9da864ac652ac
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/4.0.1 CPython/3.9.13

Release history Release notifications | RSS feed

This release

0.0.2 This release

1 release file

0.0.1

1 release file

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page