Downloads · 30 days
4
3% of all-time downloads
recommendation-models
sujal7102003/recommendation-models
recommendation-models is a machine learning model from sujal7102003. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for keras. The card lists the license as mit.
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.
All-time downloads
136
Public
Repo size
1.7 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.cbm1.1 MB · 65%
At a glance
- Library
- keras
- License
- mit
- Access
- Public
- Created
- Jun 5, 2025
- Updated
- Jun 9, 2025
- SHA
- 1b928e8a
Model card
From the Hugging Face model README
🎵 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.joblibrandom_forest.joblibxgboost.joblibsvm.joblibvoting_classifier.joblibcatboost_model.cbmann_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.
- Library
- keras
- License
- mit
- Created
- Jun 5, 2025
- Updated
- Jun 9, 2025