CodeDevX commited on
Commit
d777fcd
·
verified ·
1 Parent(s): b6f6cdb

Add model card README

Browse files
Files changed (1) hide show
  1. README.md +62 -0
README.md ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ library_name: pytorch
4
+ tags:
5
+ - time-series
6
+ - forecasting
7
+ - lstm
8
+ - multi-task
9
+ - multi-domain
10
+ pipeline_tag: time-series-forecasting
11
+ ---
12
+
13
+ # Future Prediction Models (Multi-Domain LSTM)
14
+
15
+ Trained PyTorch LSTM checkpoints forecasting 7 daily time-series domains (AI/NVIDIA, Programming/npm, Finance/BTC, Sports/ATP Elo, Weather/Chennai, Economy/S&P500, Energy/WTI).
16
+
17
+ ## Checkpoints
18
+
19
+ All models predict 7 steps from a 60-step window of base-normalized values `x_t = value_t / value_{t-1} - 1`.
20
+
21
+ | File | Parameters (hidden, layers, dropout) |
22
+ |---|---|
23
+ | `unified_model.pt` | Multi-task: shared LSTM (128, 2) + domain embedding (16) + 7 heads; domains: ai, programming, finance, sports, weather, economy, energy |
24
+ | `model_ai.pt` | 128 hidden, 3 layers, dropout 0.1 |
25
+ | `model_finance.pt` | 128 hidden, 3 layers, dropout 0.1 |
26
+ | `model_programming.pt` | 128 hidden, 3 layers, dropout 0.1 |
27
+ | `model_sports.pt` | 128 hidden, 3 layers, dropout 0.1 |
28
+ | `model_economy.pt` | 64 hidden, 2 layers, dropout 0.1 |
29
+ | `model_energy.pt` | 64 hidden, 2 layers, dropout 0.1 |
30
+ | `model_weather.pt` | 64 hidden, 2 layers, dropout 0.1 |
31
+
32
+ ## Validated accuracy (held-out, H1 MAPE)
33
+
34
+ | Topic | Separate | Unified |
35
+ |---|---|---|
36
+ | AI | 1.86% | 2.36% |
37
+ | Programming | 7.12% | 5.42% |
38
+ | Finance | 1.52% | 1.74% |
39
+ | Sports | 0.06% | 0.04% |
40
+ | Weather | 0.18% | 0.18% |
41
+ | Economy | 0.64% | 0.75% |
42
+ | Energy | 2.83% | 1.90% |
43
+
44
+ ## Model architecture (reference)
45
+
46
+ ```python
47
+ class LSTMForecaster(nn.Module):
48
+ def __init__(self, input_size=1, hidden_size=128, num_layers=3, dropout=0.1, horizon=7):
49
+ ...
50
+ self.lstm = nn.LSTM(input_size, hidden_size, num_layers, batch_first=True, dropout=...)
51
+ self.head = nn.Sequential(nn.Linear(hidden_size, hidden_size), nn.ReLU(),
52
+ nn.Dropout(dropout), nn.Linear(hidden_size, horizon))
53
+ def forward(self, x): # x: (B, 60, 1)
54
+ out, _ = self.lstm(x)
55
+ return self.head(out[:, -1, :]) # (B, 7)
56
+ ```
57
+
58
+ Forecasts are statistical estimates on validated hold-out data; no model can predict the future with 100% accuracy.
59
+
60
+ ## License
61
+
62
+ MIT