Back to all work
AI / ML / Personal project

Star Type Classification

Model comparison on 240 stars across six classes.

scikit-learnKNNRandom Forest
Star type classification across six classesClass 0Class 1Class 2Class 3Class 4Class 5240 stars / 6 classesKNN / Tree / Forest
Star type classification 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.