Tabular Classification
K2
Safetensors
PyTorch
exoplanet
transit-detection
astronomy
kepler
tess
1d-cnn
custom-model
sarojpatil16's picture
Add training script reference
2ec81d5 verified
raw
history blame contribute delete
687 Bytes
"""
Exoplanet Transit Detection — Multi-Branch 1D CNN (AstroNet/ExoMiner++ style)
==============================================================================
Based on:
- ExoMiner++ (arxiv:2502.09790, Feb 2025) — NASA Ames
- AstroNet-Triage-v2 (arxiv:2301.01371, Jan 2023) — production-deployed
Dataset: bingbangboom/exoplanet-transit-detection
- Kepler + TESS + K2 multi-mission data
- 18,853 train / 2,357 val / 2,357 test
- 3-class: PLANET, NO_SIGNAL, FALSE_POSITIVE
- Flux views: global(201), local(81), odd(201), even(201) + scalar features
Results: 89.1% accuracy, 89.0% F1 (weighted) on test set
"""
# Full training script - see train.py in repo for details