| --- |
| license: mit |
| datasets: |
| - google/speech_commands |
| base_model: |
| - facebook/wav2vec2-base |
| --- |
| # Command Classifier |
|
|
| This is a lightweight classifier model trained on top of Wav2Vec2 features for classifying speech commands. |
|
|
| - **Base model**: `facebook/wav2vec2-base` |
| - **Fine-tuned model**: Lightweight linear classifier |
| - **Dataset**: [Google Speech Commands v0.02](https://arxiv.org/abs/1804.03209) |
|
|
| ## Usage |
|
|
| You can use this classifier by combining it with Wav2Vec2 features. The classifier expects mean-pooled Wav2Vec2 hidden states. |
|
|
| ```python |
| from transformers import Wav2Vec2Model |
| from command_classifier import CommandClassifier |
| import torch |
| |
| wav2vec = Wav2Vec2Model.from_pretrained("facebook/wav2vec2-base") |
| classifier = CommandClassifier(num_classes=35) |
| classifier.load_state_dict(torch.load("pytorch_model.bin")) |
| |
