Team Ai
Modelpublic

StanislavKo28/music_moods_classification

sourceHugging Faceapache-2.0updated 6mo agoView on Hugging Face
4likes22kdownloads
Model Card

Music mood classification is a fundamental and versatile application in many various domains. Some possible use cases for music mood classification include:

  • —music recommendation systems;
  • —content organization and discovery;
  • —radio broadcasting and programming;
  • —music licensing and copyright management;
  • —music analysis and research;
  • —content tagging and metadata enrichment;
  • —audio identification and copyright protection;
  • —music production and creativity;
  • —healthcare and therapy;
  • —entertainment and gaming.

The model is trained based on publicly available dataset of labeled music data — HWNAS Dataset — that contains 6930 sample 30-second audio files evenly split among 14 moods:

  • —angry;
  • —dark;
  • —energetic;
  • —epic;
  • —euphoric;
  • —happy;
  • —mysterious;
  • —relaxing;
  • —romantic;
  • —sad;
  • —scary;
  • —glamorous;
  • —uplifting;
  • —sentimental.

Kaggle notebooks:

EpochF1Eval LossEval PrecisionEval RecallEval ROC-AUC
10.048754752008875812.6325404644012450.0305446603710686870.13440.6071502456887531
20.121884630049791332.5345952510833740.099457561280596410.20960.7306199790432439
30.129326544016288932.37766551971435550.106659096275395230.23680.7810705576029681
40.154393255359183522.25601768493652340.215726225980786560.24640.8006652336372222
50.153671261674781842.17846131324768070.233050949050949030.24960.8082513555319044
60.171849784875469422.11602425575256350.20529156194744770.24960.8182308683863416
70.211903108023807682.0695686340332030.30766896603905850.27360.8201191204703708
80.206102497434568572.03871154785156250.31118209880092020.26880.8236533161925041
90.213958259470999172.00540733337402340.242773228225031280.27520.8259563342861382
100.2313886457612971.96186137199401860.242666197277642550.280.8353918139721079
110.25516847346948291.94175279140472410.27127938431253090.29760.8357069507628714
120.238592032063360471.94679653644561770.245603063462608460.28640.8338472674725235
130.232112972201774761.92734897136688230.233045014331928650.27520.8381106641271369
140.222416285775530741.9416925907135010.229664728296185980.2720.8346930076748398
150.256096905660333661.90333735942840580.25762263976194410.29440.8394839561830068
160.247237285565444751.8890686035156250.24796162747260070.28160.8410164948564808
170.23922902311709791.89304554462432860.23345213196389670.27520.8409307834060327
180.260440614537806871.86626172065734860.25873236545314440.29440.8451350890943117
190.256759208541761.87683439254760740.25269571315831570.28320.8423786926762681
200.257034653274140551.8709082603454590.255700253651332860.28320.8437720096266398
210.26220510406486871.86933708190917970.260973459006350550.28960.8440420530281421
220.248807222641617731.86535823345184330.246990211004283870.27840.8439683504327168