opensource text summarization toolkit.
Project description
summarizers
summarizers
is package for controllable summarization based on CTRLsum.- currently, we only supports English. It doesn't work in other languages.
Installation
pip install summarizers
Usage
1. Create Summarizers
- First at all, create summarizers obejct to summarize your own article.
>>> from summarizers import Summarizers
>>> summ = Summarizers()
- You can select type of source article between [
normal
,paper
,patent
]. - If you don't input any parameter, default type is
normal
.
>>> from summarizers import Summarizers
>>> summ = Summarizers('normal') # <-- default.
>>> summ = Summarizers('paper')
>>> summ = Summarizers('patent')
- If you want GPU acceleration, set param
device='cuda'
.
>>> from summarizers import Summarizers
>>> summ = Summarizers('normal', device='cuda')
2. Basic Summarization
- If you inputted source article, basic summariztion is conducted.
>>> contents = """
Tunip is the Octonauts' head cook and gardener.
He is a Vegimal, a half-animal, half-vegetable creature capable of breathing on land as well as underwater.
Tunip is very childish and innocent, always wanting to help the Octonauts in any way he can.
He is the smallest main character in the Octonauts crew.
"""
>>> summ(contents)
'Tunip is a Vegimal, a half-animal, half-vegetable creature'
3. Query focused Summarization
- If you want to input query together, Query focused summarization conducted.
>>> summ(contents, query="main character of Octonauts")
'Tunip is the smallest main character in the Octonauts crew.'
3. Abstractive QA (Auto Question Detection)
- If you inputted question as query, Abstractive QA is conducted.
>>> summ(contents, query="What is Vegimal?")
'Half-animal, half-vegetable'
- You can turn off this feature by setting param
question_detection=False
.
>>> summ(contents, query="SOME_QUERY", question_detection=False)
4. Prompt based Summarization
- You can generate summary that begins with some sequence using param
prompt
. - It works like GPT-3's Prompt based generation. (but It doesn't work very well.)
>>> summ(contents, prompt="Q:Who is Tunip? A:")
"Q:Who is Tunip? A: Tunip is the Octonauts' head"
5. Query focused Summarization with Prompt
- You can also input both
query
andprompt
. - In this case, a query focus summary is generated that starts with a prompt.
>>> summ(contents, query="personality of Tunip", prompt="Tunip is very")
"Tunip is very childish and innocent, always wanting to help the Octonauts."
6. Options for Decoding Strategy
- For generative models, decoding strategy is very important.
summarizers
support variety of options for decoding strategy.
>>> summ(
... contents=contents,
... num_beams=10,
... top_k=30,
... top_p=0.85,
... no_repeat_ngram_size=3,
... )
License
Copyright 2021 Hyunwoong Ko.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
No source distribution files available for this release.See tutorial on generating distribution archives.
Built Distribution
File details
Details for the file summarizers-1.0.4-py3-none-any.whl
.
File metadata
- Download URL: summarizers-1.0.4-py3-none-any.whl
- Upload date:
- Size: 9.4 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.4.1 importlib_metadata/3.7.3 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.59.0 CPython/3.7.3
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 612fbd1100d3db34f1e3c6ab7624d5a3d9cc2717fca9c7e64007c389affa4a2f |
|
MD5 | 7665464fb9971a9990f862f11c1fb742 |
|
BLAKE2b-256 | da1905d1562685dce594b8f0f6ff6a61349c84b2183a1292ee93398c2233564e |