Skip to content

This repository contains a machine learning project focused on classifying potentially hazardous asteroids using orbital and physical parameters. The project addresses significant class imbalance by applying SMOTE resampling techniques and develops interpretable models using Random Forests and XGBoost. It aims to provide efficient and accurate plan

Notifications You must be signed in to change notification settings

rishimohan2715/asteroid-hazard-classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Asteroid Hazard Classification for Planetary Defense

Research Project: Lightweight Machine Learning for Near-Earth Object Risk Assessment

🎯 Project Overview

Binary classifier to predict Potentially Hazardous Asteroids (PHAs) using orbital parameters, achieving planetary defense-grade accuracy while being 100x faster than deep learning approaches.

📊 Dataset

  • Source: NASA Asteroids Classification (Kaggle)
  • Size: 90,836 asteroids
  • Features: Absolute magnitude, diameter, velocity, miss distance, eccentricity
  • Target: Binary (hazardous: 9%, non-hazardous: 91%)

🚀 Quick Start

Setup Environment

python3 -m venv asteroid_env source asteroid_env/bin/activate pip install -r requirements.txt cat > requirements.txt << 'EOF'

Core Data Science

numpy==1.24.3 pandas==2.0.3 scipy==1.11.1

Machine Learning

scikit-learn==1.3.0 xgboost==1.7.6 imbalanced-learn==0.11.0

Visualization

matplotlib==3.7.2 seaborn==0.12.2 plotly==5.15.0

Jupyter

jupyter==1.0.0 ipykernel==6.25.0 ipywidgets==8.1.0

Data Download

kaggle==1.5.16

Model Persistence

joblib==1.3.2

Utilities

tqdm==4.66.1 astropy==5.3.2 python-dotenv==1.0.0

About

This repository contains a machine learning project focused on classifying potentially hazardous asteroids using orbital and physical parameters. The project addresses significant class imbalance by applying SMOTE resampling techniques and develops interpretable models using Random Forests and XGBoost. It aims to provide efficient and accurate plan

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published