Allanatrix commited on
Commit
fb4759f
·
verified ·
1 Parent(s): ebcfe17

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +87 -3
README.md CHANGED
@@ -1,6 +1,90 @@
1
  ---
2
  license: apache-2.0
3
  pipeline_tag: tabular-classification
4
- tags:
5
- - not-for-all-audiences
6
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
  pipeline_tag: tabular-classification
4
+ ---
5
+ # NexaAstro: Stellar Classification with SDSS Data
6
+
7
+ **NexaAstro - Stellar Classification** is a hybrid machine learning model for classifying stars using data from the Sloan Digital Sky Survey (SDSS). It leverages a two-stage architecture:
8
+
9
+ - **CatBoost**: Gradient boosting for robust feature modeling.
10
+ - **Feedforward Neural Network (NN)**: Refines predictions for enhanced accuracy.
11
+
12
+ This model is part of the [Nexa Scientific Model Suite](https://huggingface.co/spaces/Allanatrix/NexaHub), dedicated to advancing scientific discovery through machine learning.
13
+
14
+ ---
15
+
16
+ ## Model Overview
17
+
18
+ - **Task**: Multi-class stellar classification (e.g., Main Sequence, White Dwarf, Giant).
19
+ - **Input**: SDSS stellar attributes (u, g, r, i, z magnitudes, spectral lines, etc.).
20
+ - **Output**: Predicted stellar class label.
21
+ - **Architecture**: CatBoost for feature extraction, followed by a Feedforward Neural Network for classification.
22
+
23
+ ---
24
+
25
+ ## Applications
26
+
27
+ - **Stellar Population Studies**: Analyzing distributions and characteristics of stellar types.
28
+ - **Galaxy Classification Support**: Providing stellar data for broader galactic studies.
29
+ - **Astrophysics Education**: Enabling interactive learning and research tools.
30
+ - **Feature Engineering**: Supporting advanced astronomical machine learning workflows.
31
+
32
+ ---
33
+
34
+ ## Getting Started
35
+
36
+ ### Example Usage
37
+ ```python
38
+ import joblib
39
+ import torch
40
+ import numpy as np
41
+ from my_nn_model import StellarNN # Replace with actual neural network module
42
+
43
+ # Load CatBoost model
44
+ catboost_model = joblib.load("Allanatrix/catboost_model.pkl")
45
+
46
+ # Load PyTorch neural network model
47
+ nn_model = StellarNN()
48
+ nn_model.load_state_dict(torch.load("Allanatrix/stellar_nn.pt"))
49
+ nn_model.eval()
50
+
51
+ # Example prediction with SDSS features
52
+ features = np.array([...]) # SDSS input features (e.g., magnitudes, spectral data)
53
+ catboost_out = catboost_model.predict(features)
54
+ refined_pred = nn_model(torch.tensor(catboost_out).float())
55
+ ```
56
+
57
+ Refer to the model documentation for detailed preprocessing and input requirements.
58
+
59
+ ---
60
+
61
+ ## Dataset
62
+
63
+ - **Source**: [Sloan Digital Sky Survey (SDSS)](https://www.sdss.org/).
64
+ - **Preprocessing**: Data cleaned, normalized, and filtered by magnitude thresholds.
65
+ - **Labels**: Discrete stellar class labels derived from expert annotations.
66
+
67
+ ---
68
+
69
+ ## Citation and License
70
+
71
+ If you use NexaAstro in your research, please cite this repository and acknowledge the SDSS dataset.
72
+ The model and associated code are licensed under the **Boost Software License 1.1 (BSL-1.1)**.
73
+
74
+ ---
75
+
76
+ ## Part of the Nexa Scientific Ecosystem
77
+
78
+ Explore related tools and models in the Nexa ecosystem:
79
+ - [Nexa Data Studio](https://huggingface.co/spaces/Allanatrix/NexaDataStudio): Tools for data processing and visualization.
80
+ - [Nexa R&D](https://huggingface.co/spaces/Allanatrix/NexaR&D): Research-focused model development environment.
81
+ - [Nexa Infrastructure](https://huggingface.co/spaces/Allanatrix/NexaInfrastructure): Scalable ML deployment solutions.
82
+ - [Nexa Hub](https://huggingface.co/spaces/Allanatrix/NexaHub): Central portal for Nexa resources.
83
+
84
+ *Coming Soon:*
85
+ - Galaxy Morphology Classifier
86
+ - Exoplanet Transit Detection Model
87
+
88
+ ---
89
+
90
+ *Developed and maintained by [Allan](https://huggingface.co/Allanatrix), an independent machine learning researcher specializing in astrophysical and scientific AI systems.*