Skip to content

Regression-based Music Emotion Prediction using Triplet Neural Networks

Notifications You must be signed in to change notification settings

AMAAI-Lab/IJCNN2020_music_emotion

 
 

Repository files navigation

IJCNN2020_music_emotion

source code for the paper publised in IJCNN 2020 "Regression-based Music Emotion Prediction using Triplet Neural Networks"

Dependencies

natsort==7.0.1

tensorflow-GPU==1.13.1

keras==2.3.1

seaborn==0.10.0

scikit-learn==0.19.2

pydub==0.23.1

matplotlib==3.0.2

tqdm

Instruction

The train the model, please refer to Training.ipynb. The model is trained on DEAM dataset, which can be downloaded here. The data processing steps can be found in DEAM/Data_Processing.ipynb.

To use your trained model to predict music emotion, please refer to Demonstration.ipynb.

Demonstration

In each of the region, we picked one song, and the predicted song can be found in /Demo/*

Low

UpperLow

Middle

LowerHigh

High

About

Regression-based Music Emotion Prediction using Triplet Neural Networks

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 98.2%
  • Python 1.8%