README.md

MusicTagging (WIP)

Introduction

General machine learning / deep learning models for music tagging.

Analysis of Weighted BCE Loss Functions for Music Tagging

Dataset: MagnaTagATune

• SVM
• FCN
• ShortChunk CNN
• RNN
• TCN
• TSN
• CRNN
• Attention

Create conda environment

conda create --name tag python=3.8
conda activate tag
pip install -r requirements.txt 

Preprocess

python preprocess.py

Training

Auto training

python autotrain.py

SVM

python train_svm.py

FCN

python train/trainer.py --config="config/config_fcn_logmel_bce.yaml"

Inference

python infer.py --model_folder="models/"

Summary all models

python generate_results.py