/music-genre-classification

Comparison of contextual (BERT) and uncontextual (GloVe and Word2Vec) word embeddings in the task of music genre classification from lyrics.

Primary LanguageJupyter NotebookMIT LicenseMIT

music-genre-classification

Easiest way to undertand the process is following the guides 1-5:

  1. Download the data from Kaggle and move it to data folder.
  2. Apply preprocessing.
  3. Baseline
  4. Training
  5. Evaluatuin

use *.ipynb files.

More documentation will be updated soon.