def transformer(model: str = 'bert', **kwargs):
"""
Load Transformer zero-shot model.
Parameters
----------
model : str, optional (default='bert')
Model architecture supported. Allowed values:
* ``'bert'`` - Google BERT BASE parameters.
* ``'tiny-bert'`` - Google BERT TINY parameters.
* ``'albert'`` - Google ALBERT BASE parameters.
* ``'tiny-albert'`` - Google ALBERT TINY parameters.
* ``'xlnet'`` - Google XLNET BASE parameters.
* ``'alxlnet'`` - Malaya ALXLNET BASE parameters.
Returns
-------
result : malaya.model.bert.ZEROSHOT_BERT class
"""
return _transformer(
model = model,
bert_class = ZEROSHOT_BERT,
xlnet_class = ZEROSHOT_XLNET,
**kwargs
)
def transformer(model: str = 'bert', **kwargs):
"""
Load Transformer zero-shot model.
Parameters
----------
model : str, optional (default='bert')
Model architecture supported. Allowed values:
* ``'bert'`` - BERT architecture from google.
* ``'tiny-bert'`` - BERT architecture from google with smaller parameters.
* ``'albert'`` - ALBERT architecture from google.
* ``'tiny-albert'`` - ALBERT architecture from google with smaller parameters.
* ``'xlnet'`` - XLNET architecture from google.
* ``'alxlnet'`` - XLNET architecture from google + Malaya.
Returns
-------
result : malaya.model.bert.ZEROSHOT_BERT class
"""
return _transformer(
model = model,
bert_class = ZEROSHOT_BERT,
xlnet_class = ZEROSHOT_XLNET,
**kwargs
)
def transformer(model: str = 'bert', quantized: bool = False, **kwargs):
"""
Load Transformer zero-shot model.
Parameters
----------
model : str, optional (default='bert')
Model architecture supported. Allowed values:
* ``'bert'`` - Google BERT BASE parameters.
* ``'tiny-bert'`` - Google BERT TINY parameters.
* ``'albert'`` - Google ALBERT BASE parameters.
* ``'tiny-albert'`` - Google ALBERT TINY parameters.
* ``'xlnet'`` - Google XLNET BASE parameters.
* ``'alxlnet'`` - Malaya ALXLNET BASE parameters.
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 : malaya.model.bert.ZEROSHOT_BERT class
"""
return _transformer(model=model,
bert_class=ZEROSHOT_BERT,
xlnet_class=ZEROSHOT_XLNET,
quantized=quantized,
siamese=False,
**kwargs)