cerebras.modelzoo.data_preparation.nlp.extractive_summarisation_utils.BertCSVFormatter#

class cerebras.modelzoo.data_preparation.nlp.extractive_summarisation_utils.BertCSVFormatter(params)[source]#

Bases: object

Converts input into bert format, sets extractive summarization targets based on the rouge score between references and input sentences. :param params: dict params: BertData configuration parameters.

Methods

process

previous

cerebras.modelzoo.data_preparation.nlp.extractive_summarisation_utils

next

cerebras.modelzoo.data_preparation.nlp.extractive_summarisation_utils.BertInputFeatures