Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/chakki-works/sumeval
Well tested & Multi-language evaluation framework for text summarization.
https://github.com/chakki-works/sumeval
bleu machine-learning rouge text-summarization
Last synced: 7 days ago
JSON representation
Well tested & Multi-language evaluation framework for text summarization.
- Host: GitHub
- URL: https://github.com/chakki-works/sumeval
- Owner: chakki-works
- License: apache-2.0
- Created: 2017-11-29T07:11:02.000Z (about 7 years ago)
- Default Branch: master
- Last Pushed: 2022-07-15T18:41:18.000Z (over 2 years ago)
- Last Synced: 2024-03-24T01:21:09.252Z (10 months ago)
- Topics: bleu, machine-learning, rouge, text-summarization
- Language: Python
- Homepage:
- Size: 108 KB
- Stars: 599
- Watchers: 19
- Forks: 62
- Open Issues: 6
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
- starred-awesome - sumeval - Well tested & Multi-language evaluation framework for text summarization. (Python)
README
Well tested & Multi-language
evaluation framework for Text Summarization.
[](https://badge.fury.io/py/sumeval)
[](https://travis-ci.org/chakki-works/sumeval)
[](https://codecov.io/gh/chakki-works/sumeval)
* Well tested
* The ROUGE-X scores are tested compare with [original Perl script (ROUGE-1.5.5.pl)](https://github.com/summanlp/evaluation).
* The BLEU score is calculated by [SacréBLEU](https://github.com/mjpost/sacrebleu), that produces the same values as official script (`mteval-v13a.pl`) used by WMT.
* Multi-language
* Not only English, Japanese and Chinese are also supported. The other language is extensible [easily](https://github.com/chakki-works/sumeval#welcome-contribution-tada).
Of course, implementation is Pure Python!
## How to use
```py
from sumeval.metrics.rouge import RougeCalculator
rouge = RougeCalculator(stopwords=True, lang="en")
rouge_1 = rouge.rouge_n(
summary="I went to the Mars from my living town.",
references="I went to Mars",
n=1)
rouge_2 = rouge.rouge_n(
summary="I went to the Mars from my living town.",
references=["I went to Mars", "It's my living town"],
n=2)
rouge_l = rouge.rouge_l(
summary="I went to the Mars from my living town.",
references=["I went to Mars", "It's my living town"])
# You need spaCy to calculate ROUGE-BE
rouge_be = rouge.rouge_be(
summary="I went to the Mars from my living town.",
references=["I went to Mars", "It's my living town"])
print("ROUGE-1: {}, ROUGE-2: {}, ROUGE-L: {}, ROUGE-BE: {}".format(
rouge_1, rouge_2, rouge_l, rouge_be
).replace(", ", "\n"))
```
```py
from sumeval.metrics.bleu import BLEUCalculator
bleu = BLEUCalculator()
score = bleu.bleu("I am waiting on the beach",
"He is walking on the beach")
bleu_ja = BLEUCalculator(lang="ja")
score_ja = bleu_ja.bleu("私はビーチで待ってる", "彼がベンチで待ってる")
```
### From the command line
```
sumeval r-nlb "I'm living New York its my home town so awesome" "My home town is awesome"
```
output.
```
{
"options": {
"stopwords": true,
"stemming": false,
"word_limit": -1,
"length_limit": -1,
"alpha": 0.5,
"input-summary": "I'm living New York its my home town so awesome",
"input-references": [
"My home town is awesome"
]
},
"averages": {
"ROUGE-1": 0.7499999999999999,
"ROUGE-2": 0.6666666666666666,
"ROUGE-L": 0.7499999999999999,
"ROUGE-BE": 0
},
"scores": [
{
"ROUGE-1": 0.7499999999999999,
"ROUGE-2": 0.6666666666666666,
"ROUGE-L": 0.7499999999999999,
"ROUGE-BE": 0
}
]
}
```
Undoubtedly you can use file input. Please see more detail by `sumeval -h`.
## Install
```
pip install sumeval
```
## Dependencies
* BLEU is depends on [SacréBLEU](https://github.com/mjpost/sacrebleu)
* To calculate `ROUGE-BE`, [`spaCy`](https://github.com/explosion/spaCy) is required.
* To use lang `ja`, [`janome`](https://github.com/mocobeta/janome) or [`MeCab`](https://github.com/taku910/mecab) is required.
* Especially to get score of `ROUGE-BE`, [`GiNZA`](https://github.com/megagonlabs/ginza) is needed additionally.
* To use lang `zh`, [`jieba`](https://github.com/fxsjy/jieba) is required.
* Especially to get score of `ROUGE-BE`, [`pyhanlp`](https://github.com/hankcs/pyhanlp) is needed additionally.
## Test
`sumeval` uses two packages to test the score.
* [pythonrouge](https://github.com/tagucci/pythonrouge)
* It calls original perl script
* `pip install git+https://github.com/tagucci/pythonrouge.git`
* [rougescore](https://github.com/bdusell/rougescore)
* It's simple python implementation for rouge score
* `pip install git+git://github.com/bdusell/rougescore.git`
## Welcome Contribution :tada:
### Add supported language
The tokenization and dependency parse process for each language is located on `sumeval/metrics/lang`.
You can make language class by inheriting [`BaseLang`](https://github.com/chakki-works/sumeval/blob/master/sumeval/metrics/lang/base_lang.py).