jcl347 commited on
Commit
4afb130
·
verified ·
1 Parent(s): f7ff494

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +21 -6
README.md CHANGED
@@ -14,14 +14,29 @@ license: mit
14
 
15
  ## Per-Stock Models
16
 
17
- **89 individual per-stock models**, each under `per_stock/{SYMBOL}.onnx`.
18
 
19
- - **Architecture**: iTransformer with RevIN (427,641 parameters each)
20
- - **Input**: 60 days x 83 features (OHLCV technicals + macro)
21
  - **Output**: 60-day forward return forecast
22
  - **Training**: Walk-forward validation (70/15/15), HuberLoss(delta=0.02)
23
- - **Average 7-day directional accuracy**: 51.3%
24
- - **Average 30-day directional accuracy**: 55.8%
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
25
 
26
  ## Usage
27
 
@@ -32,7 +47,7 @@ import numpy as np
32
  # Load per-stock model
33
  session = ort.InferenceSession("per_stock/AAPL.onnx")
34
 
35
- # features shape: (1, 60, 83)
36
  output = session.run(None, {"features": features})[0]
37
  # output shape: (1, 60) — predicted returns
38
  ```
 
14
 
15
  ## Per-Stock Models
16
 
17
+ **{len(per_stock_metrics)} individual per-stock models**, each under `per_stock/{SYMBOL}.onnx`.
18
 
19
+ - **Architecture**: iTransformer with RevIN (433,274 parameters each)
20
+ - **Input**: 60 days x 126 features (OHLCV technicals + macro + relative strength)
21
  - **Output**: 60-day forward return forecast
22
  - **Training**: Walk-forward validation (70/15/15), HuberLoss(delta=0.02)
23
+ - **Average 7-day directional accuracy**: 51.8%
24
+ - **Average 30-day directional accuracy**: 56.8%
25
+
26
+ ## v7.0 Features (126 total — includes FRED macro)
27
+
28
+ - 83 original: OHLCV technicals, momentum, volume, volatility, statistics, calendar, macro
29
+ - 4 relative strength: stock vs SPY returns (5/20/60d), SPY correlation
30
+ - 3 cross-asset: VIX correlation, rolling beta, volume-price correlation
31
+ - 4 advanced volume: MFI, A/D line, VWAP deviation, Force Index
32
+ - 5 price structure: range position, ATR ratio, consecutive days, candle body
33
+ - 5 statistical regime: Hurst exponent, Parkinson/GK volatility, consistency, tail ratio
34
+ - 4 intermarket: SPY momentum, gold/oil ratio, DXY-VIX interaction
35
+ - 3 sector ETF: stock vs sector ETF returns (5/20d), sector correlation
36
+ - 4 credit market: HYG/TLT returns, credit spread proxy, HYG-SPY divergence
37
+ - 1 VIX term structure: VIX9D/VIX short-term fear ratio
38
+ - 2 industry commodity: per-stock commodity correlation and return
39
+ - 2 intermarket extended: copper/gold ratio change, BTC sentiment
40
 
41
  ## Usage
42
 
 
47
  # Load per-stock model
48
  session = ort.InferenceSession("per_stock/AAPL.onnx")
49
 
50
+ # features shape: (1, 60, 126)
51
  output = session.run(None, {"features": features})[0]
52
  # output shape: (1, 60) — predicted returns
53
  ```