Back to work
Machine Learning · 2023

Song Virality Prediction Using ML

Predictive model analyzing 114,002 Spotify tracks to forecast song virality. Used clustering-based virality definition with K-Means and multiple classification algorithms achieving 100% accuracy with regularized models.

Song virality prediction model performance and feature analysis
[ KEY ]

Technical highlights

  1. 01

    Clustering-based virality definition using popularity, danceability, and energy features.

  2. 02

    Comparative analysis of 5 ML algorithms including Random Forest, Decision Trees, and Logistic Regression.

[ STK ]

Stack

  • Python
  • Scikit-learn
  • K-Means
  • Random Forest
  • Spotify API