cerebras.modelzoo.data_preparation.nlp.extractive_summarisation_utils.BertInputFeatures#

class cerebras.modelzoo.data_preparation.nlp.extractive_summarisation_utils.BertInputFeatures(input_token_ids, labels, segment_ids, cls_ids)#

Bases: tuple

Create new instance of BertInputFeatures(input_token_ids, labels, segment_ids, cls_ids)

Methods

count

Return number of occurrences of value.

index

Return first index of value.

Attributes

cls_ids

Alias for field number 3

input_token_ids

Alias for field number 0

labels

Alias for field number 1

segment_ids

Alias for field number 2
__add__(value, /)#

Return self+value.

__mul__(value, /)#

Return self*value.

cls_ids#

Alias for field number 3

count(value, /)#

Return number of occurrences of value.

index(value, start=0, stop=9223372036854775807, /)#

Return first index of value.

Raises ValueError if the value is not present.

input_token_ids#

Alias for field number 0

labels#

Alias for field number 1

segment_ids#

Alias for field number 2

previous

cerebras.modelzoo.data_preparation.nlp.extractive_summarisation_utils.BertCSVFormatter

next

cerebras.modelzoo.data_preparation.nlp.gpt2