π΅ 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