A prediction model that can predict how highly rated music will be based on audio characteristics and track information from Spotify.
Music is always changing, isn't it? It seems like everyone has a favorite song or artist at the center of their world that you've never even heard of! With that kind of variability, how is it possible that we have top 10 lists and "best of" compilations? In short: are there aspects of songs that make them subjectively good to lots of people? If so, can we predict a song's popularity based on some or all of its characteristics?
I think we can! Given some tendencies in music I've heard over the past few years, a few features spring to mind: "danceability", being in a major key, and probably certain telltale lyrics ("love", "c'mon"). Before cracking open any data, my uninformed opinion is that these will be somewhat predictive things to look at.
Spotify Charts is a cool little project launched by Spotify in 2013, showcasing the most listened-to and most viral tracks each week. They provide a convenient .csv that contains the track name, artist name, top 200 ranking, and a URL that points to the track on Spotify's web client.
I then used Spotipy, a convenient Python wrapper for the Spotify web API, to get the audio characteristics. The wrapper's audio_features method takes a song ID, which I extracted from the URL in the csv file. The hardest part remains: putting everything neatly in a DataFrame together such that we can make predictions off it!
This data is pristine! No cleaning or modification was needed, it went into a DataFrame as-is.
Not much exploration yet. Need a complete DataFrame first!
From Spotify's web API documentation, I gathered enough background to form a laundry list of features:
- acousticness - A confidence measure from 0.0 to 1.0 of whether the track is acoustic. 1.0 represents high confidence the track is acoustic.
- analysis_url - An HTTP URL to access the full audio analysis of this track. An access token is required to access this data.
- danceability - Danceability describes how suitable a track is for dancing based on a combination of musical elements including tempo, rhythm stability, beat strength, and overall regularity. A value of 0.0 is least danceable and 1.0 is most danceable.
- duration_ms - The duration of the track in milliseconds.
- energy - Energy is a measure from 0.0 to 1.0 and represents a perceptual measure of intensity and activity. Typically, energetic tracks * feel fast, loud, and noisy. For example, death metal has high energy, while a Bach prelude scores low on the scale. Perceptual features contributing to this attribute include dynamic range, perceived loudness, timbre, onset rate, and general entropy.
- id - The Spotify ID for the track.
- instrumentalness - Predicts whether a track contains no vocals. "Ooh" and "aah" sounds are treated as instrumental in this context. Rap or spoken word tracks are clearly "vocal". The closer the instrumentalness value is to 1.0, the greater likelihood the track contains no vocal content. Values above 0.5 are intended to represent instrumental tracks, but confidence is higher as the value approaches 1.0.
- key - The key the track is in. Integers map to pitches using standard Pitch Class notation. E.g. 0 = C, 1 = C♯/D♭, 2 = D, and so on.
- liveness - Detects the presence of an audience in the recording. Higher liveness values represent an increased probability that the track was performed live. A value above 0.8 provides strong likelihood that the track is live.
- loudness - The overall loudness of a track in decibels (dB). Loudness values are averaged across the entire track and are useful for comparing relative loudness of tracks. Loudness is the quality of a sound that is the primary psychological correlate of physical strength (amplitude). Values typical range between -60 and 0 db.
- mode - Mode indicates the modality (major or minor) of a track, the type of scale from which its melodic content is derived. Major is represented by 1 and minor is 0.
- speechiness - Speechiness detects the presence of spoken words in a track. The more exclusively speech-like the recording (e.g. talk show, audio book, poetry), the closer to 1.0 the attribute value. Values above 0.66 describe tracks that are probably made entirely of spoken words. Values between 0.33 and 0.66 describe tracks that may contain both music and speech, either in sections or layered, including such cases as rap music. Values below 0.33 most likely represent music and other non-speech-like tracks.
- tempo - The overall estimated tempo of a track in beats per minute (BPM). In musical terminology, tempo is the speed or pace of a given piece and derives directly from the average beat duration.
- time_signature - An estimated overall time signature of a track. The time signature (meter) is a notational convention to specify how many beats are in each bar (or measure).
- track_href - A link to the Web API endpoint providing full details of the track.
- type - The object type: "audio_features"
- uri - The Spotify URI for the track.
- valence - A measure from 0.0 to 1.0 describing the musical positiveness conveyed by a track. Tracks with high valence sound more positive (e.g. happy, cheerful, euphoric), while tracks with low valence sound more negative (e.g. sad, depressed, angry).
- Get everything into a DataFrame, abstracting features into dummy variables when necessary.
- Fit a model to the data.
- Gauge accuracy (at least at first, I plan on using cross-validation for this).
- Repeat steps 2-3 with different estimators to improve accuracy.
- Add new features, partly out of curiosity and partly to improve accuracy.
- (optional) dimensionality reduction.
The first challenge is the one I'm on now: getting the data into a usable format!
Spotify themselves could use this! A label or individual could also use it to estimate an up-and-coming artist's potential rank.
None yet - please hold