Update README.md

This commit is contained in:
Prithiviraj Damodaran 2021-05-08 12:35:17 +00:00 committed by huggingface-web
parent d4ad4ddaff
commit 48d2ea5ad4
1 changed files with 47 additions and 2 deletions

View File

@ -14,9 +14,18 @@ pip install git+https://github.com/PrithivirajDamodaran/Parrot.git
# Caveat: the generate part is NOT seeded, so for a same input, multiple runs will produce DIFFERENT outputs for now
from parrot import Parrot
import torch
import warnings
warnings.filterwarnings("ignore")
def set_seed(seed):
torch.manual_seed(seed)
if torch.cuda.is_available():
torch.cuda.manual_seed_all(seed)
set_seed(42)
#Init models (make sure you init ONLY once if you integrate this to your code)
parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5", use_gpu=False)
@ -33,7 +42,7 @@ for phrase in phrases:
print(para_phrase)
```
<pre>
```
-----------------------------------------------------------------------------
Input_phrase: Can you recommed some upscale restaurants in Rome?
-----------------------------------------------------------------------------
@ -55,7 +64,43 @@ Input_phrase: What are the famous places we should not miss in Russia
"what are some of the most important places to visit in russia?"
"what are some of the most famous places of russia?"
"what are some places we should not miss in russia?"
</pre>
```
### How to get syntactic and phrasal diversity/variety in your paraphrases using parrot?
You can play with the do_diverse knob (check out the next section for more knobs).
Consider this example: do_diverse = False (default)
```
------------------------------------------------------------------------------
Input_phrase: The ultimate test of your knowledge is your capacity to convey it to another.
------------------------------------------------------------------------------
'the final test of knowledge is your capacity to impart it '
'the ultimate test of a person's knowledge is his ability to transmit it to another '
'the ultimate test of knowledge is the ability to communicate it to another '
'the ultimate test of knowledge is your ability to communicate it to others '
'the test of your knowledge is your capacity to communicate it to others '
'the ultimate test for knowledge is the capacity to show it to another '
'the ultimate test of your knowledge is your ability to transmit to others '
'the ultimate test of a knowledge is your capacity to communicate it to another '
'the ultimate test of knowledge is your capacity to transmit it to another '
'the final test of your knowledge is your ability to convey it to another '
```
do_diverse = True
```
------------------------------------------------------------------------------
Input_phrase: The ultimate test of your knowledge is your capacity to convey it to another.
------------------------------------------------------------------------------
'one of the ultimate tests of knowledge is your ability to communicate it to another person '
'one of the ultimate tests of knowledge is your ability to transmit it to another person '
'one of the ultimate tests of knowledge is your ability to communicate it to another '
'one of the greatest tests of knowledge is your ability to convey it to another '
'one of the ultimate tests of knowledge is your ability to transmit it to another '
'one of the ultimate tests of knowledge is your ability to convey it to another person '
'one of the ultimate tests of knowledge is the ability to convey it to another '
'the ultimate test of your knowledge is your ability to communicate it to another '
'the ultimate test of your knowledge is your ability to transmit it to another '
```
### Knobs