Views
No views yet
| Model | RMSE | MAE |
|---|---|---|
| Random Forest | 0.0094 | 0.0070 |
| Gradient Boosting | 0.0090 | 0.0068 |
| Linear Regression | 0.0011 | 0.0008 |
| LSTM | 0.0067 | 0.0059 |
tron_sklearn_models.pkl: Scikit-learn models (RF, GB, LR)tron_scaler.pkl: Feature scalertron_lstm_model.h5: LSTM neural networktron_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="tron_sklearn_models.pkl"
9)
10scaler_path = hf_hub_download(
11 repo_id="YOUR_USERNAME/YOUR_REPO",
12 filename="tron_scaler.pkl"
13)
14lstm_path = hf_hub_download(
15 repo_id="YOUR_USERNAME/YOUR_REPO",
16 filename="tron_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)