Speakato Trainer is a project used to create models conforming to the requirements of Speakato.
Write down the desired commands, prepare dataset and Speakato will do the rest.
WARNING: Only Windows is supported!
Before you start don't forget to install necessary packages:
pip install -r requirements.txt
python -m spacy download pl_core_news_sm
python -m spacy download en_core_web_sm...or use setup.cmd to configure virutal environment with all required packages and modules!
Put the list of commands along with it's labels in the dataset.json. Data should be saved in the following form:
[
{
"text": "some text",
"command": "example_command"
}
]Save command list in the commands.txt file:
greeting
open
close
alarm
Sample datasets have been included in examples\
Prepare dataset, run python src\main.py and pass necessary arguments. That's all!
Example usage with flags:
python src/main.py --language pl --dataset .\examples\polish_commands_dataset\ --model .\models\test --mode 1
Available modes:
- 1 - Create new model
- 2 - Add new data to previously created model
You can use predefined methods from src\Sandbox.ipynb to test your model.