Skip to content

Built an ensemble model on the Spotify dataset to determine the popularity of songs and study feature importance using SHAP.

Notifications You must be signed in to change notification settings

StarrySkyrs/Spotify_Popularity_Prediction

Repository files navigation

Spotify Popularity Prediction

Built an ensemble model on the Spotify dataset to determine the popularity of songs and study feature importance using SHAP.

Models:

The ensemble model includes:

  • "ridge"
  • "random forest"
  • "XGBoost"
  • "LightGBM"
  • "CatBoost"

Predict Results:

The first result:

The predicted popularity is 50.07, which is close to true popularity 50.

pred_1 image

The second result:

The predicted popularity is 38.71, which is close to true popularity 41.

pred_2 image

About

Built an ensemble model on the Spotify dataset to determine the popularity of songs and study feature importance using SHAP.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published