lyric-tagger

The codebase for finetuning flant5 model to include tag in a lyric.

Dataset

The dataset should be in the following structure:

data/
  1.txt
  2.txt
  ...
  lyrics.json

To train the model, run the following command:

python finetune.py

To generate the dataset, run the following command:

python generate_lyric.py

To install the required packages, run the following command:

pip install -r requirements.txt

This project is licensed under the MIT License - see the LICENSE file for details.

Name		Name	Last commit message	Last commit date
Latest commit History 19 Commits
.github/workflows		.github/workflows
dataset		dataset
model		model
.gitignore		.gitignore
.pylintc		.pylintc
LICENSE		LICENSE
README.md		README.md
enums.py		enums.py
evaluate_script.py		evaluate_script.py
exceptions.py		exceptions.py
finetune.py		finetune.py
generate_lyric.py		generate_lyric.py
requirements.txt		requirements.txt
test_model.py		test_model.py