alpha-predict / src /strategy.py
DevKX's picture
Upload 14 files
4cb21eb verified
# src/strategy.py
def get_market_regime(prediction_prob):
"""
Translates model probability into actionable trading zones.
"""
if prediction_prob >= 0.578:
return {
"zone": "GREEN ZONE (Sniper)",
"color": "green",
"icon": "๐Ÿš€",
"action": "Move/Stay to 3x S&P 500 (SPXL/UPRO)",
"tag": "High Conviction Bullish"
}
elif 0.530 <= prediction_prob < 0.578:
return {
"zone": "YELLOW ZONE (Normal)",
"color": "orange",
"icon": "โš–๏ธ",
"action": "Move/Stay to 1x S&P 500 (SPY/VOO)",
"tag": "Standard Market Beta"
}
else:
return {
"zone": "RED ZONE (Cash)",
"color": "red",
"icon": "๐Ÿ›ก๏ธ",
"action": "Move/Stay to CASH (0x)",
"tag": "Risk Aversion / Defensive"
}