A PyTorch-based binary classification model that predicts whether a song is "hype" or not based on Spotify's audio features. This model was trained on Spotify audio feature data recorded April 2019 and prior: https://www.kaggle.com/datasets/tomigelo/spotify-audio-features/data
This model makes use of Spotify's audio features (energy, danceability, valence, and tempo) to classify songs into two categories:
- Hype: High-energy, danceable tracks
- Not Hype: Lower-energy, more relaxed tracks
- Built with PyTorch for flexible model architecture
- Binary classification with sigmoid output
- Input: 9 normalized audio features
- Hidden layers with ReLU activation
- Dropout for regularization
- Sigmoid output for binary classification
Audio features used and how Spotify describes them:
- Energy: Intensity and power of the track
- Danceability: How suitable the track is for dancing
- Valence: Musical positivity/happiness
- Tempo: Speed of the track (BPM)