Back To Projects
Stellar Classification based on Numerous Characteristics using Machine Learning
Roberto T. | Fall 2022 |

The task of stellar classification can be tedious and lengthy when done manually. One can expedite stellar classification by creating an artificial intelligence model to automate the process. The current stellar classification model serves to effectively categorize stars for research purposes regarding their distribution around the universe, so automating the development of this resource would allow professionals to allocate more time to explore the bounds of our current understanding of space and the universe. After finding and analyzing a dataset containing numerical and categorical features, a supervised learning approach was then used to train and test different models on their ability to classify the stars in the given test set. A Decision Tree Classifier, Random Forest Classifier, Ridge Classifier, and Support Vector Classifier were trained and tested using the data.


The task of stellar classification can be tedious and lengthy when done manually. One can expedite stellar classifi- cation by creating an artificial intelligence model to automate the process. As we as a species continue to explore the frontier of the observable universe, we should seek to automate time intensive problems like stellar classification. The current stellar classification model serves to effectively categorize stars for re- search purposes regarding their distribution around the universe, so automating the development of this resource would allow professionals to allocate more time to explore the bounds of our current understanding of space and the universe. After finding and analyzing a dataset containing numerical and categorical features, a supervised learning approach was then used to train and test different models on their ability to classify the stars in the given test set. A Decision Tree Classifier, Random Forest Classifier, Ridge Classifier, and Support Vector Classifier were trained and tested using the data. The most successful models were the Decision Tree Classifier and Random Forest Classifier, each with about a 94 percent prediction accuracy across different accuracy metrics on the test data. Despite some drawbacks in regards to the availability of usable data, four models were trained and two were proven to be consistently and successfully accurate. Any future attempts at developing models for stellar classification should concentrate more on gathering data as to have a more thoroughly trained set of models.

Explore More!

Published Paper
Roberto T.
Sophia Barton
Computer Science MS from Stanford

Related Projects

workspace_premium
Diversified AI Techniques for Augmenting Brain Tumor Diagnosis

This research explores the application of AI technology to expedite the diagnosis of brain tumors.
Dhruv M. | Fall 2022
Mentored by Odysseas Drosis
workspace_premium
Artificial Intelligence in the Stock Market: Predicting Prices

This research project focuses on stock price prediction through A.I. models as well as machine learning algorithms to maximize profit potential, improve investments, and eliminate risk.
Emin C. | Fall 2022
Mentored by Odysseas Drosis
Using Machine Learning to Classify Stars, Quasars, and Galaxies

In this project, we looked at data from stars, quasars, and galaxies from the sixteenth data release of the Sloan Digital Sky Survey Telescope. The aim of the project was to accurately and quickly classify these three types of objects using machine learning. We used three machine learning algorithms, namely logistic regression, multi-layer perceptron, and decision tree classifier.
Chinmay R. | Summer 2022
Mentored by Amanda Wang