Files
PythagorasGoal/scripts/analysis/xs01_dispersion_gate.py
T

190 lines
7.7 KiB
Python

"""XS01 dispersion-gate — la reversione cross-sectional va accesa solo in certi regimi?
Motivazione: l'edge XS01 e' concentrato (2025 domina, 2023 debole). Ipotesi da
testare: il fattore reversione cross-sezionale paga quando c'e' DISPERSIONE da
far rientrare (spread cross-section largo) e/o correlazione media alta (mosse
idiosincratiche = rumore che rientra), e perde nei regime-break (dispersione da
trend divergente, es. melt-up di un singolo asset).
Metodo (anti-multiple-testing):
[1] DIAGNOSTICA: engine XS01 canonico SENZA gate, registrando per ogni trade
il valore di 3 feature di regime alla barra di ENTRY (tutte causali,
calcolate dallo stesso panel closes <= i):
g_disp = std cross-section del segnale stesso (logC[i]-logC[i-lb])
g_corr = correlazione media pairwise 72h (identita' var dell'indice)
g_vol = vol realizzata BTC 168h
Bucket per quintili (quintili dal TRAIN) -> mean net per bucket,
TRAIN e OOS SEPARATI. Si prosegue solo se la relazione e' monotona
e con lo stesso segno in entrambe le finestre.
[2] GATE: per la feature promossa, sweep soglie (percentili TRAIN
30/40/50/60/70) -> TRAIN/OOS Sharpe/PnL/DD vs base. Serve PLATEAU.
[3] Solo se [2] regge: gate PORT06 (swap equity sleeve XS01).
uv run python scripts/analysis/xs01_dispersion_gate.py
"""
from __future__ import annotations
import sys
from pathlib import Path
import numpy as np
import pandas as pd
PROJECT_ROOT = Path(__file__).resolve().parents[2]
sys.path.insert(0, str(PROJECT_ROOT))
from scripts.strategies.XS01_cross_sectional import (
aligned_panel, UNIVERSE, FEE_RT, LEV, POS, OOS_FRAC, LB, HOLD)
N_A = len(UNIVERSE)
def build_features(M, lb=LB):
"""Feature di regime causali dal panel closes (nessun feed esterno)."""
logC = np.log(M.values)
r = np.diff(logC, axis=0, prepend=logC[:1]) # ret orari (r[0]=0)
R = pd.DataFrame(r, index=M.index)
# g_disp: std cross-section del momentum lb (il segnale che fadiamo)
D = pd.DataFrame(logC).diff(lb).to_numpy()
g_disp = np.nanstd(D, axis=1)
# g_corr 72h: avg pairwise corr via identita' della varianza dell'indice
w = 72
idx_var = R.mean(axis=1).rolling(w).var().to_numpy()
mean_var = R.rolling(w).var().mean(axis=1).to_numpy()
with np.errstate(divide="ignore", invalid="ignore"):
g_corr = (N_A * idx_var / mean_var - 1) / (N_A - 1)
# g_vol: vol BTC 168h annualizzata
b = UNIVERSE.index("BTC")
g_vol = R[b].rolling(168).std().to_numpy() * np.sqrt(24 * 365)
return dict(g_disp=g_disp, g_corr=g_corr, g_vol=g_vol)
def sim_with_trace(M, feats, gate=None, lb=LB, hold=HOLD, fee_rt=FEE_RT,
lev=LEV, pos=POS):
"""Engine XS01 canonico (stessa logica/ordine di XS01_cross_sectional.xsec_sim)
+ trace per-trade (entry index, net, feature) + gate opzionale bool[i]."""
C = M.values
ts = pd.to_datetime(M.index, unit="ms", utc=True)
n = len(C)
logC = np.log(C)
cap = peak = 1000.0
dd = 0.0
rows = []
eq_ts, eq_v = [], []
last = -1
i = lb
fee = 2 * fee_rt
while i < n - hold:
if i <= last:
i += 1
continue
if gate is not None and not gate[i]:
i += 1
continue
dm = (logC[i] - logC[i - lb])
dm = dm - dm.mean()
w = -dm
gw = np.sum(np.abs(w))
if gw < 1e-9:
i += 1
continue
w = w / gw
book = float(np.sum(w * (logC[i + hold] - logC[i])))
net = book - fee
cap = max(cap + cap * pos * lev * net, 10.0)
peak = max(peak, cap)
dd = max(dd, (peak - cap) / peak)
rows.append((i, int(ts[i].year), net,
feats["g_disp"][i], feats["g_corr"][i], feats["g_vol"][i]))
eq_ts.append(ts[i + hold])
eq_v.append(cap)
last = i + hold
i += 1
tr = pd.DataFrame(rows, columns=["i", "year", "net", "g_disp", "g_corr", "g_vol"])
yrs_span = (ts[-1] - ts[0]).days / 365.25 or 1
out = dict(trades=len(tr), cap=cap, dd=dd * 100, eq_ts=eq_ts, eq_v=eq_v, tr=tr)
if len(tr) > 1 and tr["net"].std() > 0:
out["sharpe"] = float(tr["net"].mean() / tr["net"].std()
* np.sqrt(len(tr) / yrs_span))
else:
out["sharpe"] = 0.0
out["pnl_add"] = float(tr["net"].sum() * 100) if len(tr) else 0.0
out["win"] = float((tr["net"] > 0).mean() * 100) if len(tr) else 0.0
out["tpm"] = len(tr) / (yrs_span * 12)
return out
def metrics_window(tr, lo, hi, yrs_span):
t = tr[(tr["i"] >= lo) & (tr["i"] < hi)]
if len(t) < 2 or t["net"].std() == 0:
return dict(n=len(t), pnl=0.0, sh=0.0, win=0.0)
sh = float(t["net"].mean() / t["net"].std() * np.sqrt(len(t) / yrs_span))
return dict(n=len(t), pnl=float(t["net"].sum() * 100), sh=sh,
win=float((t["net"] > 0).mean() * 100))
def main():
M = aligned_panel()
n = len(M)
cut = int(n * (1 - OOS_FRAC))
ts = pd.to_datetime(M.index, unit="ms", utc=True)
feats = build_features(M)
print("=" * 96)
print(f" XS01 dispersion-gate | panel {ts[0].date()} -> {ts[-1].date()} "
f"({n} ore, 8 asset) | TRAIN 70% (-> {ts[cut].date()}) / OOS 30%")
print("=" * 96)
base = sim_with_trace(M, feats)
tr = base["tr"]
yrs_tr = (ts[cut] - ts[0]).days / 365.25
yrs_oo = (ts[-1] - ts[cut]).days / 365.25
# [1] DIAGNOSTICA per quintili (quintili dal TRAIN)
print("\n[1] DIAGNOSTICA — mean net per trade (bps) per quintile feature @entry")
ttr = tr[tr["i"] < cut]
too = tr[tr["i"] >= cut]
for g in ("g_disp", "g_corr", "g_vol"):
qs = ttr[g].quantile([0.2, 0.4, 0.6, 0.8]).to_numpy()
def bucket(x):
return int(np.searchsorted(qs, x))
print(f" {g:<7s} | " + " | ".join(
f"Q{q+1} TR {ttr[ttr[g].apply(bucket) == q]['net'].mean()*1e4:+6.1f} "
f"OOS {too[too[g].apply(bucket) == q]['net'].mean()*1e4:+6.1f}"
for q in range(5)) +
f" (n TR {len(ttr)}, OOS {len(too)})")
# [2] GATE sweep — per ogni feature, tieni SOPRA o SOTTO il percentile
print("\n[2] GATE — TRAIN/OOS vs base (soglie = percentili del TRAIN; "
"side scelto dal segno della diagnostica TRAIN)")
b_tr = metrics_window(tr, 0, cut, yrs_tr)
b_oo = metrics_window(tr, cut, n, yrs_oo)
print(f" {'BASE':<24s} TRAIN n {b_tr['n']:>4} pnl {b_tr['pnl']:>+7.1f}% "
f"Sh {b_tr['sh']:>5.2f} | OOS n {b_oo['n']:>4} pnl {b_oo['pnl']:>+7.1f}% "
f"Sh {b_oo['sh']:>5.2f}")
for g in ("g_disp", "g_corr", "g_vol"):
# segno dal TRAIN: correlazione quintile->ret
qs5 = ttr[g].quantile([0.2, 0.4, 0.6, 0.8]).to_numpy()
means = [ttr[ttr[g].apply(lambda x: int(np.searchsorted(qs5, x))) == q]["net"].mean()
for q in range(5)]
side = "above" if means[-1] > means[0] else "below"
for pct in (30, 40, 50, 60, 70):
thr = float(np.nanpercentile(feats[g][:cut], pct))
gv = feats[g]
gate = (gv >= thr) if side == "above" else (gv <= thr)
gate = np.nan_to_num(gate, nan=False).astype(bool)
r = sim_with_trace(M, feats, gate=gate)
g_tr = metrics_window(r["tr"], 0, cut, yrs_tr)
g_oo = metrics_window(r["tr"], cut, n, yrs_oo)
print(f" {g} {side} p{pct:<3d}{'':<6s} TRAIN n {g_tr['n']:>4} "
f"pnl {g_tr['pnl']:>+7.1f}% Sh {g_tr['sh']:>5.2f} | "
f"OOS n {g_oo['n']:>4} pnl {g_oo['pnl']:>+7.1f}% Sh {g_oo['sh']:>5.2f}")
# breakdown annuale base (riferimento concentrazione)
print("\n base, net additivo per anno (%):",
{int(y): round(float(v * 100), 1)
for y, v in tr.groupby("year")["net"].sum().items()})
if __name__ == "__main__":
main()