Python PyTorch Wav2Vec2ASRBundle.get_labels用法及代码示例

本文简要介绍python语言中 torchaudio.pipelines.Wav2Vec2ASRBundle.get_labels 的用法。

用法:

get_labels(*, bos: str = '<s>', pad: str = '<pad>', eos: str = '</s>', unk: str = '<unk>') → Tuple[str]

参数：

• bos(str,可选的) -句子标记的开头。 (默认：'<s>')

• pad(str,可选的) -填充令牌。 (默认：'<pad>')

• eos(str,可选的) -句末标记。 (默认：'</s>')

• unk(str,可选的) -未知类的令牌。 (默认：'<unk>')

返回：

对于 ASR 上的模型 fine-tuned，返回表示输出类标签的字符串元组。

返回类型：

元组[str]

输出类标签(仅适用于fine-tuned 包)

前四个代币是 BOS、padding、EOS 和 UNK 代币，它们可以自定义。

示例

>>> import torchaudio
>>> torchaudio.models.HUBERT_ASR_LARGE.get_labels()
('<s>', '<pad>', '</s>', '<unk>', '|', 'E', 'T', 'A', 'O', 'N', 'I', 'H', 'S', 'R', 'D', 'L', 'U', 'M', 'W', 'C', 'F', 'G', 'Y', 'P', 'B', 'V', 'K', "'", 'X', 'J', 'Q', 'Z')