Skip to content

transformer_mc

allennlp_models.mc.predictors.transformer_mc

[SOURCE]


TransformerMCPredictor#

@Predictor.register("transformer_mc")
class TransformerMCPredictor(Predictor)

Predictor for the TransformerMC model.

predict#

class TransformerMCPredictor(Predictor):
 | ...
 | def predict(self, prefix: str, alternatives: List[str]) -> JsonDict