Source code for malaya.zero_shot.classification

from malaya.model.bert import ZeroshotBERT
from malaya.model.xlnet import ZeroshotXLNET
from herpetologist import check_type
from malaya.similarity import available_transformer as _available_transformer, _transformer


[docs]def available_transformer(): """ List available transformer zero-shot models. """ return _available_transformer()
[docs]@check_type def transformer(model: str = 'bert', quantized: bool = False, **kwargs): """ Load Transformer zero-shot model. Parameters ---------- model: str, optional (default='bert') Check available models at `malaya.similarity.available_transformer()`. quantized: bool, optional (default=False) if True, will load 8-bit quantized model. Quantized model not necessary faster, totally depends on the machine. Returns ------- result: model List of model classes: * if `bert` in model, will return `malaya.model.bert.ZeroshotBERT`. * if `xlnet` in model, will return `malaya.model.xlnet.ZeroshotXLNET`. """ return _transformer( model=model, bert_model=ZeroshotBERT, xlnet_model=ZeroshotXLNET, quantized=quantized, siamese=False, **kwargs )