Update README.md only
Browse files
README.md
CHANGED
|
@@ -1,3 +1,49 @@
|
|
| 1 |
---
|
| 2 |
-
license:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
+
license: mit
|
| 3 |
+
tags:
|
| 4 |
+
- recommendation
|
| 5 |
+
- binary-classification
|
| 6 |
+
- music
|
| 7 |
+
- ensemble
|
| 8 |
---
|
| 9 |
+
|
| 10 |
+
# 🎵 Music Recommendation System
|
| 11 |
+
|
| 12 |
+
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.
|
| 13 |
+
|
| 14 |
+
## 📁 Files Included
|
| 15 |
+
|
| 16 |
+
- `data.csv` — Dataset of 195 songs with features like danceability, energy, loudness, tempo, etc.
|
| 17 |
+
- Trained model files:
|
| 18 |
+
- `logistic_regression.joblib`
|
| 19 |
+
- `random_forest.joblib`
|
| 20 |
+
- `xgboost.joblib`
|
| 21 |
+
- `svm.joblib`
|
| 22 |
+
- `voting_classifier.joblib`
|
| 23 |
+
- `catboost_model.cbm`
|
| 24 |
+
- `ann_model.keras`
|
| 25 |
+
- `final_model_card_scaled.pdf` — Full model evaluation, comparison table, and chart
|
| 26 |
+
|
| 27 |
+
## 🧠 Models Used
|
| 28 |
+
|
| 29 |
+
- Logistic Regression
|
| 30 |
+
- Random Forest
|
| 31 |
+
- XGBoost
|
| 32 |
+
- Support Vector Machine (SVM)
|
| 33 |
+
- Voting Classifier (Ensemble)
|
| 34 |
+
- CatBoost
|
| 35 |
+
- Artificial Neural Network (ANN)
|
| 36 |
+
|
| 37 |
+
## 📊 Evaluation
|
| 38 |
+
|
| 39 |
+
All models were evaluated using:
|
| 40 |
+
- Accuracy
|
| 41 |
+
- Precision
|
| 42 |
+
- Recall
|
| 43 |
+
- F1-Score
|
| 44 |
+
|
| 45 |
+
Refer to the PDF `final_model_card_scaled.pdf` for full details.
|
| 46 |
+
|
| 47 |
+
## 📬 Contact
|
| 48 |
+
|
| 49 |
+
Maintained by Sujal Thakkar.
|