Lyric Luminary

This is a fun project that aims to view lyrics via the lens of a text classification model. I'm experimenting with TF-IDF(and hopefully BERT) to classify songs into a genere via the lyrics of the song.

Setup and Install

Prerequisites

• Go
• Wails
• Python3
• Node

git clone https://github.com/angel-721/lyric-luminary.git
cd app
wails build
cd build/bin
./lyric-luminary

MAKE SURE YOU HAVE A .env with your API keys. THIS MAY OR MAY NOT WORK ON WINDOWS.

Training a Model

1. Use the nim scripts to break down this csv into txt files
2. Run the main.py with the arg of train python main.py trainbash

Model specs

• Tfidf Vectorizer
• Naive Bayes Classifier
• Stop words