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.
|