Views
No views yet
| Model | RMSE | MAE |
|---|---|---|
| Random Forest | 0.0217 | 0.0175 |
| Gradient Boosting | 0.0221 | 0.0183 |
| Linear Regression | 0.0019 | 0.0015 |
| LSTM | 0.0136 | 0.0105 |
stellar_sklearn_models.pkl: Scikit-learn models (RF, GB, LR)stellar_scaler.pkl: Feature scalerstellar_lstm_model.h5: LSTM neural networkstellar_metadata.json: Training metadata1from huggingface_hub import hf_hub_download
2import joblib
3from tensorflow.keras.models import load_model
4
5# Download models
6sklearn_path = hf_hub_download(
7 repo_id="YOUR_USERNAME/YOUR_REPO",
8 filename="stellar_sklearn_models.pkl"
9)
10scaler_path = hf_hub_download(
11 repo_id="YOUR_USERNAME/YOUR_REPO",
12 filename="stellar_scaler.pkl"
13)
14lstm_path = hf_hub_download(
15 repo_id="YOUR_USERNAME/YOUR_REPO",
16 filename="stellar_lstm_model.h5"
17)
18
19# Load models
20models = joblib.load(sklearn_path)
21scaler = joblib.load(scaler_path)
22lstm = load_model(lstm_path)
23
24# Make predictions
25# (prepare your features first)
26predictions = models['RandomForest'].predict(scaled_features)