AI / ML / Personal project
Star Type Classification
Model comparison on 240 stars across six classes.
The problem
Reliable comparisons on a small dataset.
The approach
Compared KNN, Decision Tree and Random Forest with scaling, stratified k-fold and learning curves.
Results and scope
- Evaluated models with cross-validation and learning curves.