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

Suggest about the index order of extractive results #68

Closed PolarisRisingWar closed 2 years ago

PolarisRisingWar commented 2 years ago

I suggest to sort the index order before print the results of extractive models.
Before the line 1155 of https://github.com/HHousen/TransformerSum/blob/master/src/extractive.py (for i in selected_ids:), I suggest to add a sort function: selected_ids.sort()
In this way, we can print the results in their original order.