modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.utils#

Classes

BertCSVFormatter

Converts input into bert format, sets extractive summarization targets based on the rouge score between references and input sentences.

BertInputFeatures

BertInputFeatures(input_token_ids, labels, segment_ids, cls_ids)