Skip to content

masked_language_model

allennlp_models.lm.predictors.masked_language_model

[SOURCE]


MaskedLanguageModelPredictor#

@Predictor.register("masked_language_model")
class MaskedLanguageModelPredictor(Predictor)

predict#

class MaskedLanguageModelPredictor(Predictor):
 | ...
 | def predict(self, sentence_with_masks: str) -> JsonDict

predictions_to_labeled_instances#

class MaskedLanguageModelPredictor(Predictor):
 | ...
 | @overrides
 | def predictions_to_labeled_instances(
 |     self,
 |     instance: Instance,
 |     outputs: Dict[str, numpy.ndarray]
 | )