🎡 Music Recommendation System

This repository hosts a collection of machine learning models designed to recommend songs by predicting whether a user is likely to "like" a track based on its audio features.

πŸ“ Files Included

  • data.csv β€” Dataset of 195 songs with features like danceability, energy, loudness, tempo, etc.
  • Trained model files:
    • logistic_regression.joblib
    • random_forest.joblib
    • xgboost.joblib
    • svm.joblib
    • voting_classifier.joblib
    • catboost_model.cbm
    • ann_model.keras
  • final_model_card_scaled.pdf β€” Full model evaluation, comparison table, and chart

🧠 Models Used

  • Logistic Regression
  • Random Forest
  • XGBoost
  • Support Vector Machine (SVM)
  • Voting Classifier (Ensemble)
  • CatBoost
  • Artificial Neural Network (ANN)

πŸ“Š Evaluation

All models were evaluated using:

  • Accuracy
  • Precision
  • Recall
  • F1-Score

Refer to the PDF final_model_card_scaled.pdf for full details.

πŸ“¬ Contact

Maintained by Sujal Thakkar.

Downloads last month
12
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support