stellar-classification

Project information

A production-grade machine learning system for classifying celestial objects — stars, galaxies, and quasars — from 500,000 photometric observations sourced from the Sloan Digital Sky Survey (SDSS DR18). The pipeline covers LOF-based outlier detection, SMOTEENN class balancing, and SelectKBest feature selection, with six classifier configurations benchmarked against each other. Random Forest achieved the highest accuracy at ~99.51%, while LightGBM was selected for deployment due to its faster inference, smaller footprint, and clean ONNX export path. The system is fully containerized with Docker, backed by a CI/CD pipeline, and served live via a Gradio interface on Hugging Face Spaces.

Python scikit-learn onnxruntime gradio docker-compose Docker pytest lightgbm