Skip to content

About

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

Topics

Resources

Stars

0 stars

Watchers

1 watching

Forks

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

0 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages