🚀 XLS-R-1B-ITALIAN-DOC4LM-5GRAM
本项目是一个针对意大利语语音识别进行微调的模型,基于XLS - R 1B模型,能有效处理意大利语语音识别任务,提升识别的准确性和效率。
🚀 快速开始
使用此模型时,请确保语音输入的采样率为16kHz。
✨ 主要特性
- 基于[facebook/wav2vec2 - xls - r - 1b](https://huggingface.co/facebook/wav2vec2 - xls - r - 1b)模型在意大利语上进行微调。
- 使用了[Common Voice 8.0](https://huggingface.co/datasets/mozilla - foundation/common_voice_8_0)、Multilingual TEDx、Multilingual LibriSpeech和Voxpopuli的训练和验证集。
- 语言模型使用意大利维基百科文章、广播报纸和电视节目的手动转录数据集生成。
📦 安装指南
下载意大利语的CommonVoice8.0数据集
from datasets import load_dataset
dataset = load_dataset("mozilla-foundation/common_voice_8_0", "it", use_auth_token=True)
💻 使用示例
基础用法
from datasets import load_dataset
dataset = load_dataset("mozilla-foundation/common_voice_8_0", "it", use_auth_token=True)
高级用法
python eval.py --model_id radiogroup-crits/wav2vec2-xls-r-1b-italian-doc4lm-5gram --dataset mozilla-foundation/common_voice_8_0 --config it --split test --log_outputs --greedy
mv log_mozilla-foundation_common_voice_8_0_it_test_predictions.txt log_mozilla-foundation_common_voice_8_0_it_test_predictions_greedy.txt
mv log_mozilla-foundation_common_voice_8_0_it_test_targets.txt log_mozilla-foundation_common_voice_8_0_it_test_targets_greedy.txt
mv mozilla-foundation_common_voice_8_0_it_test_eval_results.txt mozilla-foundation_common_voice_8_0_it_test_eval_results_greedy.txt
python eval.py --model_id radiogroup-crits/wav2vec2-xls-r-1b-italian-doc4lm-5gram --dataset mozilla-foundation/common_voice_8_0 --config it --split test --log_outputs
mv log_mozilla-foundation_common_voice_8_0_it_test_predictions.txt log_mozilla-foundation_common_voice_8_0_it_test_predictions_lm.txt
mv log_mozilla-foundation_common_voice_8_0_it_test_targets.txt log_mozilla-foundation_common_voice_8_0_it_test_targets_lm.txt
mv mozilla-foundation_common_voice_8_0_it_test_eval_results.txt mozilla-foundation_common_voice_8_0_it_test_eval_results_lm.txt
📚 详细文档
模型信息
评估结果
任务 |
数据集 |
指标 |
值 |
语音识别 |
Common Voice 8.0意大利语 |
测试WER |
9.04 |
语音识别 |
Common Voice 8.0意大利语 |
测试CER |
2.2 |
语音识别 |
Common Voice 8.0意大利语 |
测试WER (+LM) |
6.24 |
语音识别 |
Common Voice 8.0意大利语 |
测试CER (+LM) |
1.67 |
引用信息
如果您想引用此模型,可以使用以下内容:
@misc{crits2022wav2vec2-xls-r-1b-italian-doc4lm-5gram,
title={XLS-R Wav2Vec2 Italian by radiogroup crits},
author={Teraoni Prioletti Raffaele, Casagranda Paolo and Russo Francesco},
publisher={Hugging Face},
journal={Hugging Face Hub},
howpublished={\url{https://huggingface.co/radiogroup-crits/wav2vec2-xls-r-1b-italian-doc4lm-5gram}},
year={2022}
}
📄 许可证
本项目采用Apache - 2.0许可证。