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')