tfr.extension.premade.TFRBertScorer

Univariate BERT-based scorer.

Inherits From: UnivariateScorer, Scorer

View aliases

Main aliases

tfr.extension.premade.tfrbert_task.TFRBertScorer

tfr.extension.premade.TFRBertScorer(
    encoder: tf.keras.Model,
    bert_output_dropout: float,
    name: str = 'tfrbert',
    **kwargs
)

Methods

__call__

View source

__call__(
    context_features: tfr.keras.model.TensorDict,
    example_features: tfr.keras.model.TensorDict,
    mask: tf.Tensor
) -> Union[tf.Tensor, tfr.keras.model.TensorDict]

See Scorer.