Skip to content

Classification machine learning algorithm for take-home interview

Notifications You must be signed in to change notification settings

NFeruch/ML_SpotifyGenrePrediction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 

Repository files navigation

MachineLearning-SpotifyGenrePrediction

Machine learning classification take-home project for an interview with Thrasio, a mass Amazon FBA Buyer

Background:

  • Spotify has an internal tool that gives songs various attributes like dancibility, loudness, popularity, etc.
  • Given a dataset of songs with these attributes, we are interested in predicting the genre of the song.
  • Accuracy was the most important metric to take into consideration

In the Notebook:

  1. EDA and Feature Engineering
  2. Model training and hyper-parameter tuning
  3. Model accuracy interpretation

I utilized the Random Forest and Logistic Regression for classification, and received a max accuracy of >92%

About

Classification machine learning algorithm for take-home interview

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published