Allow prediction on input sentences

HHousen / TransformerSum

Models to perform neural summarization (extractive and abstractive) using machine learning transformers and a tool to convert abstractive summarization datasets to the extractive task.

https://transformersum.rtfd.io

GNU General Public License v3.0

429 stars 58 forks source link

Allow prediction on input sentences #31

Closed salman1993 closed 4 years ago

salman1993 commented 4 years ago

Changes:

Separate predict_sentences from predict which works on input text.
Return list when raw_scores=True instead of dict. This is because we might have the same text such as "Thank you." appear in multiple places in the text. However, their score might be different since they are at different positions/context. Using a dict overwrites the previous position's score.

HHousen commented 4 years ago

Alright, looks good to be merged on my end. @salman1993 Everything look good to you?

salman1993 commented 4 years ago

Yeah! Looks good, lets merge it. :)