├── .gitattributes ├── .gitignore ├── README.md ├── models └── ponswon.pth ├── preprocessing ├── msd │ ├── 50tagList.txt │ ├── 7D_id_to_path.pkl │ ├── MSD_id_to_7D_id.pkl │ ├── README.md │ ├── filtered_list_test.cP │ ├── filtered_list_train.cP │ ├── msd_id_to_tag_vector.cP │ ├── preprocess.py │ └── run.sh └── mtat │ ├── .1.py │ ├── preprocess.py │ └── split.py ├── requirements.txt ├── training ├── back_end │ ├── bert_attention.py │ ├── pons_module.py │ └── won_module.py ├── config.py ├── data_loader │ ├── msd_loader.py │ └── mtat_loader.py ├── eval.py ├── eval.sh ├── front_end │ ├── lee_conv.py │ ├── lee_module.py │ ├── pons_conv.py │ └── pons_module.py ├── main.py ├── model.py ├── setup.py ├── solver.py └── train.sh └── visualize ├── vismodel.py └── visualize_example.ipynb /.gitattributes: -------------------------------------------------------------------------------- 1 | *.* linguist-language=Python 2 | -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/README.md -------------------------------------------------------------------------------- /models/ponswon.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/models/ponswon.pth -------------------------------------------------------------------------------- /preprocessing/msd/50tagList.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/preprocessing/msd/50tagList.txt -------------------------------------------------------------------------------- /preprocessing/msd/7D_id_to_path.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/preprocessing/msd/7D_id_to_path.pkl -------------------------------------------------------------------------------- /preprocessing/msd/MSD_id_to_7D_id.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/preprocessing/msd/MSD_id_to_7D_id.pkl -------------------------------------------------------------------------------- /preprocessing/msd/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/preprocessing/msd/README.md -------------------------------------------------------------------------------- /preprocessing/msd/filtered_list_test.cP: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/preprocessing/msd/filtered_list_test.cP -------------------------------------------------------------------------------- /preprocessing/msd/filtered_list_train.cP: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/preprocessing/msd/filtered_list_train.cP -------------------------------------------------------------------------------- /preprocessing/msd/msd_id_to_tag_vector.cP: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/preprocessing/msd/msd_id_to_tag_vector.cP -------------------------------------------------------------------------------- /preprocessing/msd/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/preprocessing/msd/preprocess.py -------------------------------------------------------------------------------- /preprocessing/msd/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/preprocessing/msd/run.sh -------------------------------------------------------------------------------- /preprocessing/mtat/.1.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /preprocessing/mtat/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/preprocessing/mtat/preprocess.py -------------------------------------------------------------------------------- /preprocessing/mtat/split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/preprocessing/mtat/split.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/requirements.txt -------------------------------------------------------------------------------- /training/back_end/bert_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/training/back_end/bert_attention.py -------------------------------------------------------------------------------- /training/back_end/pons_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/training/back_end/pons_module.py -------------------------------------------------------------------------------- /training/back_end/won_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/training/back_end/won_module.py -------------------------------------------------------------------------------- /training/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/training/config.py -------------------------------------------------------------------------------- /training/data_loader/msd_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/training/data_loader/msd_loader.py -------------------------------------------------------------------------------- /training/data_loader/mtat_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/training/data_loader/mtat_loader.py -------------------------------------------------------------------------------- /training/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/training/eval.py -------------------------------------------------------------------------------- /training/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/training/eval.sh -------------------------------------------------------------------------------- /training/front_end/lee_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/training/front_end/lee_conv.py -------------------------------------------------------------------------------- /training/front_end/lee_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/training/front_end/lee_module.py -------------------------------------------------------------------------------- /training/front_end/pons_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/training/front_end/pons_conv.py -------------------------------------------------------------------------------- /training/front_end/pons_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/training/front_end/pons_module.py -------------------------------------------------------------------------------- /training/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/training/main.py -------------------------------------------------------------------------------- /training/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/training/model.py -------------------------------------------------------------------------------- /training/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/training/setup.py -------------------------------------------------------------------------------- /training/solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/training/solver.py -------------------------------------------------------------------------------- /training/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/training/train.sh -------------------------------------------------------------------------------- /visualize/vismodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/visualize/vismodel.py -------------------------------------------------------------------------------- /visualize/visualize_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/minzwon/self-attention-music-tagging/HEAD/visualize/visualize_example.ipynb --------------------------------------------------------------------------------