Models to perform neural summarization (extractive and abstractive) using machine learning transformers and a tool to convert abstractive summarization datasets to the extractive task.
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.
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.