Community Scripts

# ╔══════════════════════════════════════════════════════════════╗
# ║                  STRATEGY REQUEST LOG                       ║
# ╚══════════════════════════════════════════════════════════════╝
# Generated        : 2026-05-06 02:32:18
# Model            : XGBoost
# Feature Eng.     : SMA (20,50,200), BB (20,2.0), RSI 14, MACD (12,26,9), ATR 14 + Auto-add features: ON
# Signal / Entry   : Enter when model confidence > threshold; exit on opposite signal or SL/TP
# Optimization     : Maximize risk-adjusted return
# Risk Mgmt        : Stop loss 0.5%, Take profit 1.0%
# Risk Filter      : —
# ══════════════════════════════════════════════════════════════

# ============================================================
# AUDUSD 15-min XGBoost Strategy
# SMA + RSI + MACD + Bollinger Bands + ATR Feature Set
# Optimized for Risk-Adjusted Return
# ============================================================

# SECTION 0 — IMPORTS & CONSTANTS
import numpy as np
import pandas as pd

DATA_PATH = "/root/Desktop/QuantifyMe/data/ohlc/AUDUSD_15min.parquet"
START_DATE = "2025-04-24"
END_DATE   = "2026-04-24"
VALIDATION_DATE = ""
TRAIN_SPLIT = 0.7

# SECTION 1 — FEATURE ENGINEERING
def feature_engineering(df, close, open_, high, low):

    # ── SMA 20, 50, 200 + distance from close ──────────────────────────────
    for period in [20, 50, 200]:
        sma = close.rolling(period).mean()
        df[f"sma_{period}"] = sma
        df[f"dm_sma_{period}"] = (close - sma) / sma

    # ── Bollinger Bands (20, 2.0) ───────────────────────────────────────────
    bb_mid   = close.rolling(20).mean()
    bb_std   = close.rolling(20).std(ddof=0)
    bb_upper = bb_mid + 2.0 * bb_std
    bb_lower = bb_mid - 2.0 * bb_std
    df["bb_mid"]   = bb_mid
    df["bb_upper"] = bb_upper
    df["bb_lower"] = bb_lower
    df["bb_width"] = (bb_upper - bb_lower) / bb_mid
    bb_range       = bb_upper - bb_lower
    df["bb_pct"]   = (close - bb_lower) / bb_range

    # ── RSI 14 ──────────────────────────────────────────────────────────────
    delta  = close.diff()
    gain   = delta.clip(lower=0)
    loss   = (-delta).clip(lower=0)
    avg_gain = gain.ewm(com=13, min_periods=14).mean()
    avg_loss = loss.ewm(com=13, min_periods=14).mean()
    rs         = avg_gain / avg_loss.replace(0, np.nan)
    df["rsi_14"] = 100.0 - (100.0 / (1.0 + rs))

    # ── MACD (12, 26, 9) ────────────────────────────────────────────────────
    ema_12     = close.ewm(span=12, adjust=False).mean()
    ema_26     = close.ewm(span=26, adjust=False).mean()
    macd_line  = ema_12 - ema_26
    signal_line = macd_line.ewm(span=9, adjust=False).mean()
    df["macd_line"]   = macd_line
    df["macd_signal"] = signal_line
    df["macd_hist"]   = macd_line - signal_line

    # ── ATR 14 + Normalised ATR ─────────────────────────────────────────────
    prev_close = close.shift(1)
    tr = pd.concat([
        high - low,
        (high - prev_close).abs(),
        (low  - prev_close).abs()
    ], axis=1).max(axis=1)
    atr = tr.ewm(com=13, min_periods=14).mean()
    df["atr_14"] = atr
    df["natr"]   = atr / close

    # ── Price momentum / rate-of-change ────────────────────────────────────
    for n in [1, 4, 8, 16]:
        df[f"roc_{n}"] = close.pct_change(n)

    # ── Candle body & wick features ─────────────────────────────────────────
    body        = (close - open_).abs()
    candle_range = (high - low).replace(0, np.nan)
    df["body_ratio"]  = body / candle_range
    df["upper_wick"]  = (high  - pd.concat([close, open_], axis=1).max(axis=1)) / candle_range
    df["lower_wick"]  = (pd.concat([close, open_], axis=1).min(axis=1) - low)   / candle_range
    df["candle_dir"]  = np.where(close >= open_, 1.0, -1.0)

    # ── Volume (if present) ─────────────────────────────────────────────────
    if "volume" in df.columns:
        vol_ma = df["volume"].rolling(20).mean()
        df["vol_ratio"] = df["volume"] / vol_ma.replace(0, np.nan)

    # ── Lagged RSI & MACD histogram ─────────────────────────────────────────
    for lag in [1, 2, 3]:
        df[f"rsi_14_lag{lag}"]   = df["rsi_14"].shift(lag)
        df[f"macd_hist_lag{lag}"] = df["macd_hist"].shift(lag)

    # ── RSI overbought / oversold zones ─────────────────────────────────────
    df["rsi_ob"]     = np.where(df["rsi_14"] > 70, 1.0, 0.0)
    df["rsi_os"]     = np.where(df["rsi_14"] < 30, 1.0, 0.0)
    df["rsi_mid_up"] = np.where((df["rsi_14"] > 50) & (df["rsi_14"] <= 70), 1.0, 0.0)
    df["rsi_mid_dn"] = np.where((df["rsi_14"] >= 30) & (df["rsi_14"] < 50), 1.0, 0.0)

    # ── MACD cross signals ───────────────────────────────────────────────────
    df["macd_cross_up"] = np.where(
        (df["macd_line"] > df["macd_signal"]) &
        (df["macd_line"].shift(1) <= df["macd_signal"].shift(1)),
        1.0, 0.0
    )
    df["macd_cross_dn"] = np.where(
        (df["macd_line"] < df["macd_signal"]) &
        (df["macd_line"].shift(1) >= df["macd_signal"].shift(1)),
        1.0, 0.0
    )

    # ── Price position relative to SMA alignment ────────────────────────────
    df["trend_aligned_bull"] = np.where(
        (close > df["sma_20"]) & (df["sma_20"] > df["sma_50"]) & (df["sma_50"] > df["sma_200"]),
        1.0, 0.0
    )
    df["trend_aligned_bear"] = np.where(
        (close < df["sma_20"]) & (df["sma_20"] < df["sma_50"]) & (df["sma_50"] < df["sma_200"]),
        1.0, 0.0
    )

    # ── Bollinger Band squeeze (low volatility) ──────────────────────────────
    bb_width_ma = df["bb_width"].rolling(20).mean()
    df["bb_squeeze"] = np.where(df["bb_width"] < bb_width_ma, 1.0, 0.0)

    # ── Rolling close statistics ─────────────────────────────────────────────
    df["close_zscore_20"] = (close - close.rolling(20).mean()) / close.rolling(20).std(ddof=0)
    df["close_zscore_50"] = (close - close.rolling(50).mean()) / close.rolling(50).std(ddof=0)

    # ── Volatility regime ────────────────────────────────────────────────────
    natr_ma = df["natr"].rolling(20).mean()
    df["vol_regime_high"] = np.where(df["natr"] > natr_ma, 1.0, 0.0)

    # ── Fill NaN from indicator warm-up ─────────────────────────────────────
    df = df.bfill().ffill()

    return df


# SECTION 2 — STRATEGY CONFIG
def strategy_config():
    return {
        "title":            "AUD/USD XGBoost SMA+RSI+MACD+BB Momentum",
        "model_type":       "XGBClassifier",
        "model_params":     {
            "n_estimators":       500,
            "max_depth":          4,
            "learning_rate":      0.04,
            "subsample":          0.75,
            "colsample_bytree":   0.70,
            "min_child_weight":   5,
            "gamma":              0.1,
            "reg_alpha":          0.1,
            "reg_lambda":         1.5,
            "scale_pos_weight":   1.0,
            "objective":          "binary:logistic",
            "tree_method":        "hist",
            "random_state":       42,
        },
        "signal_threshold": 0.55,
        "direction":        "both",
        "stop_loss":        0.005,
        "take_profit":      0.010,
        "cooldown":         0,
        "max_positions":    1,
        "on_opposite":      "reverse",
        "session_filter":   [0, 23],
        "min_atr":          None,
        "trend_filter":     None,
        "target_horizon":   4,
        "objective":        (
            "Maximize risk-adjusted return (Sharpe/Calmar) on AUD/USD 15-min. "
            "XGBoost with depth-4 trees and conservative regularization (reg_lambda=1.5, "
            "min_child_weight=5) to reduce overfitting on FX data. "
            "2:1 RR (SL=0.5%, TP=1.0%) ensures positive expectancy with ~40%+ win rate. "
            "Subsample + colsample add stochastic diversity. 500 estimators with lr=0.04 "
            "balances bias-variance. Threshold 0.55 filters marginal signals."
        ),
        "notes":            (
            "Features: SMA(20/50/200) with distances, Bollinger Bands width+pct, RSI-14 "
            "with zone flags, MACD histogram + crosses, ATR-14 + NATR, momentum ROC(1/4/8/16), "
            "candle body/wick ratios, trend alignment flags, BB squeeze, z-scores, vol regime. "
            "Reverse on opposite signal to capture trend reversals. Session filter disabled "
            "to capture AUD/USD Asian + London + NY sessions. Target horizon = 4 bars (1 hour)."
        ),
    }

# ╔══════════════════════════════════════════════════════════════╗
# ║                  STRATEGY REQUEST LOG                       ║
# ╚══════════════════════════════════════════════════════════════╝
# Generated        : 2026-05-06 01:40:19
# Model            : XGBoost
# Feature Eng.     : EMA (9,21), RSI 14 + Auto-add features: ON
# Signal / Entry   : Enter when model confidence > threshold; exit on opposite signal or SL/TP
# Optimization     : Maximize risk-adjusted return
# Risk Mgmt        : Stop loss 0.5%, Take profit 1.0%
# Risk Filter      : —
# ══════════════════════════════════════════════════════════════

# ============================================================
# SECTION 0 — IMPORTS & CONSTANTS
import numpy as np
import pandas as pd

DATA_PATH = "/root/Desktop/QuantifyMe/data/ohlc/AUDUSD_15min.parquet"
START_DATE = "2025-04-24"
END_DATE   = "2026-04-24"
VALIDATION_DATE = ""
TRAIN_SPLIT = 0.7

# SECTION 1 — FEATURE ENGINEERING
def feature_engineering(df, close, open_, high, low):
    # ── EMA 9 and EMA 21 ──────────────────────────────────────────────────
    ema_9  = close.ewm(span=9,  adjust=False).mean()
    ema_21 = close.ewm(span=21, adjust=False).mean()

    df["ema_9"]    = ema_9
    df["ema_21"]   = ema_21
    df["dm_ema_9"]  = (close - ema_9)  / ema_9
    df["dm_ema_21"] = (close - ema_21) / ema_21

    # EMA crossover signal: positive when fast > slow
    df["ema_cross"]       = ema_9 - ema_21
    df["ema_cross_prev"]  = df["ema_cross"].shift(1)
    # Binary: did a cross just occur?
    df["ema_cross_up"]    = np.where((df["ema_cross"] > 0) & (df["ema_cross_prev"] <= 0), 1, 0)
    df["ema_cross_down"]  = np.where((df["ema_cross"] < 0) & (df["ema_cross_prev"] >= 0), 1, 0)
    # Trend direction encoded as -1 / 1
    df["ema_trend"]       = np.where(ema_9 > ema_21, 1, -1)

    # ── RSI 14 ────────────────────────────────────────────────────────────
    delta  = close.diff()
    gain   = delta.clip(lower=0)
    loss   = (-delta).clip(lower=0)
    avg_gain = gain.ewm(com=13, adjust=False).mean()
    avg_loss = loss.ewm(com=13, adjust=False).mean()
    rs     = avg_gain / avg_loss.replace(0, np.nan)
    rsi_14 = 100 - (100 / (1 + rs))
    df["rsi_14"] = rsi_14

    # RSI regime flags
    df["rsi_oversold"]   = np.where(rsi_14 < 30, 1, 0)
    df["rsi_overbought"] = np.where(rsi_14 > 70, 1, 0)
    df["rsi_mid"]        = rsi_14 - 50          # centred

    # RSI momentum (1-bar change in RSI)
    df["rsi_delta"]      = rsi_14.diff(1)
    df["rsi_delta2"]     = rsi_14.diff(3)

    # ── Additional momentum / volatility features ─────────────────────────
    # ATR-like normalised true range
    prev_close = close.shift(1)
    tr = pd.concat([
        high - low,
        (high - prev_close).abs(),
        (low  - prev_close).abs()
    ], axis=1).max(axis=1)
    atr_14 = tr.ewm(span=14, adjust=False).mean()
    df["atr_14"]  = atr_14
    df["natr_14"] = atr_14 / close          # normalised ATR

    # Rate-of-change over various horizons
    for n in [4, 8, 16]:
        df[f"roc_{n}"] = close.pct_change(n)

    # Bollinger Band width and %B (using 20-period SMA)
    sma_20   = close.rolling(20).mean()
    std_20   = close.rolling(20).std()
    bb_upper = sma_20 + 2 * std_20
    bb_lower = sma_20 - 2 * std_20
    df["bb_width"] = (bb_upper - bb_lower) / sma_20
    df["bb_pct"]   = (close - bb_lower) / (bb_upper - bb_lower).replace(0, np.nan)

    # Candle body and wick features
    df["body"]        = (close - open_) / close
    df["upper_wick"]  = (high - pd.concat([close, open_], axis=1).max(axis=1)) / close
    df["lower_wick"]  = (pd.concat([close, open_], axis=1).min(axis=1) - low)  / close

    # Volume-normalised momentum proxy: price range relative to ATR
    df["range_vs_atr"] = (high - low) / atr_14.replace(0, np.nan)

    # Lagged EMA cross signal
    df["ema_cross_lag1"] = df["ema_cross"].shift(1)
    df["ema_cross_lag2"] = df["ema_cross"].shift(2)

    # Combined signal: RSI and EMA cross alignment
    df["rsi_ema_bull"] = np.where((rsi_14 > 50) & (ema_9 > ema_21), 1, 0)
    df["rsi_ema_bear"] = np.where((rsi_14 < 50) & (ema_9 < ema_21), 1, 0)

    # Hour-of-day (cyclical encoding) — no lookahead
    hour = df.index.hour
    df["hour_sin"] = np.sin(2 * np.pi * hour / 24)
    df["hour_cos"] = np.cos(2 * np.pi * hour / 24)

    # Day-of-week (cyclical encoding)
    dow = df.index.dayofweek
    df["dow_sin"] = np.sin(2 * np.pi * dow / 5)
    df["dow_cos"] = np.cos(2 * np.pi * dow / 5)

    # Fill NaN from warm-up periods
    df = df.bfill().ffill()

    return df

# SECTION 2 — STRATEGY CONFIG
def strategy_config():
    return {
        "title":            "AUD/USD EMA Cross (9/21) + RSI14 XGBoost Scalper",
        "model_type":       "XGBClassifier",
        "model_params":     {
            "n_estimators":     400,
            "max_depth":        4,
            "learning_rate":    0.04,
            "subsample":        0.8,
            "colsample_bytree": 0.75,
            "min_child_weight": 3,
            "gamma":            0.1,
            "reg_alpha":        0.05,
            "reg_lambda":       1.5,
            "objective":        "binary:logistic",
            "tree_method":      "hist",
            "random_state":     42,
        },
        "signal_threshold": 0.54,
        "direction":        "both",
        "stop_loss":        0.005,
        "take_profit":      0.010,
        "cooldown":         0,
        "max_positions":    1,
        "on_opposite":      "reverse",
        "session_filter":   [0, 23],
        "min_atr":          None,
        "trend_filter":     None,
        "target_horizon":   4,
        "objective":        (
            "Maximise risk-adjusted return on AUD/USD 15-min bars. "
            "XGBoost chosen for its ability to capture non-linear interactions between "
            "the EMA-cross regime, RSI momentum, volatility (NATR/BB width), and time-of-day. "
            "Shallow trees (max_depth=4) with strong regularisation (reg_lambda=1.5, gamma=0.1) "
            "reduce overfitting on the limited 1-year window. "
            "2:1 R:R (SL=0.5%, TP=1.0%) improves Sharpe; reverse on opposite signal captures "
            "trend momentum without missing transitions."
        ),
        "notes":            (
            "Features: EMA-9/21 cross and distances, RSI-14 with regime flags and delta, "
            "ATR-14, NATR, Bollinger Band width/%B, 4/8/16-bar ROC, candle anatomy, "
            "time cyclical encodings. Threshold 0.54 filters marginal signals to raise precision. "
            "No session filter applied — AUD/USD has meaningful moves across Asian and London sessions."
        ),
    }

# ╔══════════════════════════════════════════════════════════════╗
# ║                  STRATEGY REQUEST LOG                       ║
# ╚══════════════════════════════════════════════════════════════╝
# Generated        : 2026-05-06 01:44:26
# Model            : XGBoost
# Feature Eng.     : BB (20,2.0), ATR 14 + Auto-add features: ON
# Signal / Entry   : Enter when model confidence > threshold; exit on opposite signal or SL/TP
# Optimization     : Maximize risk-adjusted return
# Risk Mgmt        : Stop loss 0.5%, Take profit 1.0%
# Risk Filter      : —
# ══════════════════════════════════════════════════════════════

# ============================================================
# SECTION 0 — IMPORTS & CONSTANTS
import numpy as np
import pandas as pd

DATA_PATH = "/root/Desktop/QuantifyMe/data/ohlc/AUDUSD_15min.parquet"
START_DATE = "2025-04-24"
END_DATE   = "2026-04-24"
VALIDATION_DATE = ""
TRAIN_SPLIT = 0.7

# SECTION 1 — FEATURE ENGINEERING
def feature_engineering(df, close, open_, high, low):
    # ── Bollinger Bands (20, 2) ──────────────────────────────────────────────
    bb_period = 20
    bb_std    = 2.0

    bb_mid   = close.rolling(bb_period).mean()
    bb_sigma = close.rolling(bb_period).std(ddof=0)
    bb_upper = bb_mid + bb_std * bb_sigma
    bb_lower = bb_mid - bb_std * bb_sigma

    df["bb_mid"]   = bb_mid
    df["bb_upper"] = bb_upper
    df["bb_lower"] = bb_lower
    df["bb_width"] = (bb_upper - bb_lower) / bb_mid
    # guard against zero range
    bb_range = bb_upper - bb_lower
    df["bb_pct"]   = np.where(bb_range != 0, (close - bb_lower) / bb_range, 0.5)

    # ── ATR (14) & Normalised ATR ────────────────────────────────────────────
    atr_period = 14
    prev_close = close.shift(1)
    tr = pd.concat([
        high - low,
        (high - prev_close).abs(),
        (low  - prev_close).abs()
    ], axis=1).max(axis=1)

    atr  = tr.ewm(span=atr_period, min_periods=atr_period, adjust=False).mean()
    df["atr"]  = atr
    df["natr"] = np.where(close != 0, atr / close, 0.0)

    # ── RSI (14) ─────────────────────────────────────────────────────────────
    rsi_period = 14
    delta  = close.diff()
    gain   = delta.clip(lower=0)
    loss   = (-delta).clip(lower=0)
    avg_gain = gain.ewm(span=rsi_period, min_periods=rsi_period, adjust=False).mean()
    avg_loss = loss.ewm(span=rsi_period, min_periods=rsi_period, adjust=False).mean()
    rs  = avg_gain / avg_loss.replace(0, np.nan)
    rsi = 100 - (100 / (1 + rs))
    df["rsi_14"] = rsi

    # ── MACD (12, 26, 9) ─────────────────────────────────────────────────────
    ema12       = close.ewm(span=12, adjust=False).mean()
    ema26       = close.ewm(span=26, adjust=False).mean()
    macd_line   = ema12 - ema26
    signal_line = macd_line.ewm(span=9, adjust=False).mean()
    df["macd"]        = macd_line
    df["macd_signal"] = signal_line
    df["macd_hist"]   = macd_line - signal_line

    # ── EMA trend features ───────────────────────────────────────────────────
    ema_20  = close.ewm(span=20, adjust=False).mean()
    ema_50  = close.ewm(span=50, adjust=False).mean()
    ema_200 = close.ewm(span=200, adjust=False).mean()
    df["ema_20"]        = ema_20
    df["ema_50"]        = ema_50
    df["ema_200"]       = ema_200
    df["close_vs_ema20"]  = (close - ema_20) / ema_20
    df["close_vs_ema50"]  = (close - ema_50) / ema_50
    df["ema20_vs_ema50"]  = (ema_20 - ema_50) / ema_50
    df["ema50_vs_ema200"] = (ema_50 - ema_200) / ema_200

    # ── Price momentum / rate-of-change ──────────────────────────────────────
    for n in [1, 4, 8, 16]:
        df[f"roc_{n}"] = close.pct_change(n)

    # ── Rolling volatility ────────────────────────────────────────────────────
    df["vol_10"]  = close.pct_change().rolling(10).std()
    df["vol_20"]  = close.pct_change().rolling(20).std()
    df["vol_ratio"] = np.where(df["vol_20"] != 0,
                               df["vol_10"] / df["vol_20"], 1.0)

    # ── Stochastic %K / %D (14, 3) ───────────────────────────────────────────
    low_14  = low.rolling(14).min()
    high_14 = high.rolling(14).max()
    stoch_range = high_14 - low_14
    stoch_k = np.where(stoch_range != 0,
                       100 * (close - low_14) / stoch_range, 50.0)
    df["stoch_k"] = stoch_k
    df["stoch_d"] = pd.Series(stoch_k, index=close.index).rolling(3).mean()

    # ── Candle body / wick features ──────────────────────────────────────────
    body      = (close - open_).abs()
    total_rng = (high - low).replace(0, np.nan)
    df["body_ratio"]  = body / total_rng
    df["upper_wick"]  = (high - pd.concat([close, open_], axis=1).max(axis=1)) / total_rng
    df["lower_wick"]  = (pd.concat([close, open_], axis=1).min(axis=1) - low) / total_rng
    df["candle_dir"]  = np.sign(close - open_)

    # ── BB interaction features ───────────────────────────────────────────────
    df["price_above_bb_upper"] = np.where(close > bb_upper, 1, 0)
    df["price_below_bb_lower"] = np.where(close < bb_lower, 1, 0)
    df["bb_squeeze"]           = np.where(df["bb_width"] < df["bb_width"].rolling(50).quantile(0.20), 1, 0)

    # ── RSI regime bins (replacing pd.cut) ───────────────────────────────────
    df["rsi_oversold"]  = np.where(rsi < 30, 1, 0)
    df["rsi_overbought"] = np.where(rsi > 70, 1, 0)
    df["rsi_neutral"]   = np.where((rsi >= 30) & (rsi <= 70), 1, 0)

    # ── Volume proxy (if volume column exists) ───────────────────────────────
    if "volume" in df.columns:
        vol_ma = df["volume"].rolling(20).mean()
        df["volume_ratio"] = np.where(vol_ma != 0,
                                      df["volume"] / vol_ma, 1.0)

    # ── Lagged features (1-bar lag to avoid lookahead) ───────────────────────
    for feat in ["bb_pct", "rsi_14", "macd_hist", "natr", "stoch_k"]:
        df[f"{feat}_lag1"] = df[feat].shift(1)
        df[f"{feat}_lag2"] = df[feat].shift(2)

    # ── Fill NaNs from warm-up ────────────────────────────────────────────────
    df = df.bfill().ffill()

    return df


# SECTION 2 — STRATEGY CONFIG
def strategy_config():
    return {
        "title":            "AUD/USD Bollinger + ATR Mean-Rev (XGBoost)",
        "model_type":       "XGBClassifier",
        "model_params": {
            "n_estimators":       400,
            "max_depth":          4,
            "learning_rate":      0.04,
            "subsample":          0.75,
            "colsample_bytree":   0.70,
            "min_child_weight":   3,
            "gamma":              0.10,
            "reg_alpha":          0.10,
            "reg_lambda":         1.50,
            "objective":          "binary:logistic",
            "random_state":       42,
            "n_jobs":             -1,
        },
        "signal_threshold": 0.56,
        "direction":        "both",
        "stop_loss":        0.005,
        "take_profit":      0.010,
        "cooldown":         0,
        "max_positions":    1,
        "on_opposite":      "reverse",
        "session_filter":   [0, 23],
        "min_atr":          None,
        "trend_filter":     None,
        "target_horizon":   4,
        "objective": (
            "Maximize risk-adjusted return (Sharpe). "
            "XGBoost with moderate depth and heavy regularisation "
            "(gamma, alpha, lambda) prevents overfit on AUD/USD 15-min data. "
            "Bollinger Bands capture mean-reversion; ATR normalises volatility; "
            "RSI + MACD confirm momentum; 2:1 TP:SL ratio supports positive expectancy."
        ),
        "notes": (
            "Features: BB (20,2) width/pct, ATR-14/NATR, RSI-14, MACD histogram, "
            "EMA 20/50/200 spreads, Stochastic %K/%D, candle-body ratios, "
            "ROC at multiple horizons, volatility ratio, BB squeeze flag, "
            "lagged versions of key features. "
            "Threshold 0.56 filters marginal signals, improving precision. "
            "target_horizon=4 (1 hour) balances signal frequency vs. noise."
        ),
    }

# ╔══════════════════════════════════════════════════════════════╗
# ║                  STRATEGY REQUEST LOG                       ║
# ╚══════════════════════════════════════════════════════════════╝
# Generated        : 2026-05-06 03:05:53
# Model            : Gradient Boosting
# Feature Eng.     : RSI 14, MACD (12,26,9) + Auto-add features: ON
# Signal / Entry   : Enter when model confidence > threshold; exit on opposite signal or SL/TP
# Optimization     : Maximize risk-adjusted return
# Risk Mgmt        : Stop loss 0.5%, Take profit 1.0%
# Risk Filter      : —
# ══════════════════════════════════════════════════════════════

# ============================================================
# SECTION 0 — IMPORTS & CONSTANTS
import numpy as np
import pandas as pd

DATA_PATH = "/root/Desktop/QuantifyMe/data/ohlc/AUDUSD_15min.parquet"
START_DATE = "2025-04-24"
END_DATE   = "2026-04-24"
VALIDATION_DATE = ""
TRAIN_SPLIT = 0.7

# SECTION 1 — FEATURE ENGINEERING
def feature_engineering(df, close, open_, high, low):
    # ── RSI 14 ──────────────────────────────────────────────────────────────
    period = 14
    delta = close.diff()
    gain = delta.clip(lower=0)
    loss = -delta.clip(upper=0)
    avg_gain = gain.ewm(com=period - 1, min_periods=period).mean()
    avg_loss = loss.ewm(com=period - 1, min_periods=period).mean()
    rs = avg_gain / avg_loss.replace(0, np.nan)
    df["rsi_14"] = 100 - (100 / (1 + rs))

    # ── MACD (12, 26, 9) ────────────────────────────────────────────────────
    ema_fast = close.ewm(span=12, adjust=False).mean()
    ema_slow = close.ewm(span=26, adjust=False).mean()
    macd_line = ema_fast - ema_slow
    signal_line = macd_line.ewm(span=9, adjust=False).mean()
    macd_hist = macd_line - signal_line
    df["macd_line"] = macd_line
    df["macd_signal"] = signal_line
    df["macd_hist"] = macd_hist

    # ── RSI derived features ─────────────────────────────────────────────────
    df["rsi_14_lag1"] = df["rsi_14"].shift(1)
    df["rsi_14_lag2"] = df["rsi_14"].shift(2)
    df["rsi_14_delta"] = df["rsi_14"] - df["rsi_14_lag1"]
    df["rsi_ob"] = np.where(df["rsi_14"] > 70, 1, 0)
    df["rsi_os"] = np.where(df["rsi_14"] < 30, 1, 0)

    # ── MACD derived features ────────────────────────────────────────────────
    df["macd_hist_lag1"] = macd_hist.shift(1)
    df["macd_hist_delta"] = macd_hist - macd_hist.shift(1)
    df["macd_cross_bull"] = np.where((macd_line > signal_line) & (macd_line.shift(1) <= signal_line.shift(1)), 1, 0)
    df["macd_cross_bear"] = np.where((macd_line < signal_line) & (macd_line.shift(1) >= signal_line.shift(1)), 1, 0)
    df["macd_above_zero"] = np.where(macd_line > 0, 1, 0)
    df["macd_hist_positive"] = np.where(macd_hist > 0, 1, 0)

    # ── ATR (14) ─────────────────────────────────────────────────────────────
    prev_close = close.shift(1)
    tr = pd.concat([
        high - low,
        (high - prev_close).abs(),
        (low - prev_close).abs()
    ], axis=1).max(axis=1)
    atr_14 = tr.ewm(span=14, adjust=False).mean()
    df["atr_14"] = atr_14
    df["natr_14"] = atr_14 / close

    # ── Bollinger Bands (20, 2) ───────────────────────────────────────────────
    bb_mid = close.rolling(20).mean()
    bb_std = close.rolling(20).std()
    bb_upper = bb_mid + 2 * bb_std
    bb_lower = bb_mid - 2 * bb_std
    df["bb_pct_b"] = (close - bb_lower) / (bb_upper - bb_lower).replace(0, np.nan)
    df["bb_width"] = (bb_upper - bb_lower) / bb_mid.replace(0, np.nan)
    df["bb_squeeze"] = np.where(df["bb_width"] < df["bb_width"].rolling(50).mean(), 1, 0)

    # ── Price momentum features ───────────────────────────────────────────────
    df["ret_1"] = close.pct_change(1)
    df["ret_4"] = close.pct_change(4)
    df["ret_8"] = close.pct_change(8)
    df["ret_16"] = close.pct_change(16)

    # ── Rolling volatility ────────────────────────────────────────────────────
    df["vol_8"] = df["ret_1"].rolling(8).std()
    df["vol_20"] = df["ret_1"].rolling(20).std()

    # ── EMA trend features ───────────────────────────────────────────────────
    ema_20 = close.ewm(span=20, adjust=False).mean()
    ema_50 = close.ewm(span=50, adjust=False).mean()
    df["ema_20"] = ema_20
    df["ema_50"] = ema_50
    df["price_vs_ema20"] = (close - ema_20) / ema_20
    df["price_vs_ema50"] = (close - ema_50) / ema_50
    df["ema20_vs_ema50"] = (ema_20 - ema_50) / ema_50

    # ── Candle body / wick features ──────────────────────────────────────────
    body = (close - open_).abs()
    candle_range = (high - low).replace(0, np.nan)
    df["body_ratio"] = body / candle_range
    df["upper_wick"] = (high - pd.concat([close, open_], axis=1).max(axis=1)) / candle_range
    df["lower_wick"] = (pd.concat([close, open_], axis=1).min(axis=1) - low) / candle_range
    df["bull_candle"] = np.where(close > open_, 1, 0)

    # ── Rolling high/low position ─────────────────────────────────────────────
    roll_high_20 = high.rolling(20).max()
    roll_low_20 = low.rolling(20).min()
    roll_range_20 = (roll_high_20 - roll_low_20).replace(0, np.nan)
    df["price_position_20"] = (close - roll_low_20) / roll_range_20

    # ── RSI + MACD interaction ────────────────────────────────────────────────
    df["rsi_macd_product"] = df["rsi_14"] * macd_hist
    df["rsi_norm"] = (df["rsi_14"] - 50) / 50

    # ── Session / time features ───────────────────────────────────────────────
    if hasattr(close.index, "hour"):
        df["hour"] = close.index.hour
        df["session_london"] = np.where((close.index.hour >= 7) & (close.index.hour < 16), 1, 0)
        df["session_ny"] = np.where((close.index.hour >= 13) & (close.index.hour < 21), 1, 0)
        df["session_asia"] = np.where((close.index.hour >= 22) | (close.index.hour < 7), 1, 0)
    else:
        df["hour"] = 0
        df["session_london"] = 0
        df["session_ny"] = 0
        df["session_asia"] = 0

    # ── Fill NaN from warm-up ────────────────────────────────────────────────
    df = df.bfill().ffill()

    return df

# SECTION 2 — STRATEGY CONFIG
def strategy_config():
    return {
        "title": "AUD/USD RSI+MACD Gradient Boosting Scalper",
        "model_type": "GradientBoostingClassifier",
        "model_params": {
            "n_estimators": 400,
            "max_depth": 4,
            "learning_rate": 0.05,
            "subsample": 0.8,
            "min_samples_leaf": 20,
            "max_features": "sqrt",
            "validation_fraction": 0.1,
            "n_iter_no_change": 30,
            "tol": 1e-4,
        },
        "signal_threshold": 0.55,
        "direction": "both",
        "stop_loss": 0.005,
        "take_profit": 0.01,
        "cooldown": 0,
        "max_positions": 1,
        "on_opposite": "reverse",
        "session_filter": None,
        "min_atr": None,
        "trend_filter": None,
        "target_horizon": 4,
        "objective": (
            "Maximize risk-adjusted return on AUD/USD 15-min data using GradientBoostingClassifier. "
            "RSI-14 captures momentum extremes and divergence conditions; MACD (12,26,9) provides "
            "trend direction and momentum shifts via crossovers and histogram slope. Additional "
            "features (BB, ATR, EMA trend, candle structure, session timing) enrich the feature "
            "space. GradientBoosting with shallow trees (depth=4), moderate learning rate (0.05), "
            "and early stopping via n_iter_no_change prevents overfitting. SL=0.5%, TP=1.0% gives "
            "a 1:2 risk/reward ratio, improving Sharpe and Calmar. Threshold=0.55 filters low-confidence "
            "signals to reduce noise. Reverse on opposite signal maximizes capital efficiency."
        ),
        "notes": (
            "n_estimators=400 with early stopping balances bias-variance. max_depth=4 keeps trees "
            "shallow to reduce overfitting on FX microstructure noise. subsample=0.8 adds stochastic "
            "gradient boosting regularization. min_samples_leaf=20 prevents fitting to outlier bars. "
            "max_features='sqrt' adds feature randomization similar to random forests. "
            "target_horizon=4 (1 hour) aligns with typical AUD/USD intraday swing durations."
        ),
    }

# ╔══════════════════════════════════════════════════════════════╗
# ║                  STRATEGY REQUEST LOG                       ║
# ╚══════════════════════════════════════════════════════════════╝
# Generated        : 2026-05-06 02:21:53
# Model            : Gradient Boosting
# Feature Eng.     : EMA (9,21), RSI 14 + Auto-add features: ON
# Signal / Entry   : Enter when model confidence > threshold; exit on opposite signal or SL/TP
# Optimization     : Maximize risk-adjusted return
# Risk Mgmt        : Stop loss 0.5%, Take profit 1.0%
# Risk Filter      : —
# ══════════════════════════════════════════════════════════════

# ============================================================
# SECTION 0 — IMPORTS & CONSTANTS
import numpy as np
import pandas as pd

DATA_PATH = "/root/Desktop/QuantifyMe/data/ohlc/AUDUSD_15min.parquet"
START_DATE = "2025-04-24"
END_DATE   = "2026-04-24"
VALIDATION_DATE = ""
TRAIN_SPLIT = 0.7

# SECTION 1 — FEATURE ENGINEERING
def feature_engineering(df, close, open_, high, low):
    # --- EMA 9 and EMA 21 (required) ---
    ema_9 = close.ewm(span=9, adjust=False).mean()
    ema_21 = close.ewm(span=21, adjust=False).mean()

    df["ema_9"] = ema_9
    df["ema_21"] = ema_21
    df["dm_ema_9"] = (close - ema_9) / ema_9
    df["dm_ema_21"] = (close - ema_21) / ema_21

    # EMA crossover signal and spread
    df["ema_cross"] = ema_9 - ema_21
    df["ema_cross_prev"] = df["ema_cross"].shift(1)
    df["ema_cross_sign"] = np.sign(df["ema_cross"])
    df["ema_cross_change"] = df["ema_cross_sign"] - np.sign(df["ema_cross_prev"])

    # --- RSI 14 (required) ---
    delta = close.diff()
    gain = delta.clip(lower=0)
    loss = -delta.clip(upper=0)
    avg_gain = gain.ewm(com=13, adjust=False).mean()
    avg_loss = loss.ewm(com=13, adjust=False).mean()
    rs = avg_gain / avg_loss.replace(0, np.nan)
    rsi_14 = 100 - (100 / (1 + rs))
    df["rsi_14"] = rsi_14

    # RSI derived features
    df["rsi_14_norm"] = (rsi_14 - 50) / 50
    df["rsi_overbought"] = np.where(rsi_14 > 70, 1, 0)
    df["rsi_oversold"] = np.where(rsi_14 < 30, 1, 0)
    df["rsi_mid_cross"] = np.where(rsi_14 > 50, 1, -1)

    # --- Additional EMAs for context ---
    ema_50 = close.ewm(span=50, adjust=False).mean()
    ema_200 = close.ewm(span=200, adjust=False).mean()
    df["ema_50"] = ema_50
    df["ema_200"] = ema_200
    df["dm_ema_50"] = (close - ema_50) / ema_50
    df["dm_ema_200"] = (close - ema_200) / ema_200
    df["ema_50_200_spread"] = (ema_50 - ema_200) / ema_200

    # --- ATR (14 periods) ---
    tr1 = high - low
    tr2 = (high - close.shift(1)).abs()
    tr3 = (low - close.shift(1)).abs()
    true_range = pd.concat([tr1, tr2, tr3], axis=1).max(axis=1)
    atr_14 = true_range.ewm(com=13, adjust=False).mean()
    df["atr_14"] = atr_14
    df["natr_14"] = atr_14 / close

    # --- Bollinger Bands (20, 2) ---
    sma_20 = close.rolling(20).mean()
    std_20 = close.rolling(20).std()
    bb_upper = sma_20 + 2 * std_20
    bb_lower = sma_20 - 2 * std_20
    df["bb_upper"] = bb_upper
    df["bb_lower"] = bb_lower
    df["bb_pct_b"] = (close - bb_lower) / (bb_upper - bb_lower).replace(0, np.nan)
    df["bb_width"] = (bb_upper - bb_lower) / sma_20

    # --- MACD (12, 26, 9) ---
    ema_12 = close.ewm(span=12, adjust=False).mean()
    ema_26 = close.ewm(span=26, adjust=False).mean()
    macd_line = ema_12 - ema_26
    macd_signal = macd_line.ewm(span=9, adjust=False).mean()
    macd_hist = macd_line - macd_signal
    df["macd_line"] = macd_line
    df["macd_signal_line"] = macd_signal
    df["macd_hist"] = macd_hist
    df["macd_hist_sign"] = np.sign(macd_hist)
    df["macd_hist_change"] = np.sign(macd_hist) - np.sign(macd_hist.shift(1))

    # --- Momentum & Rate of Change ---
    df["mom_5"] = close.pct_change(5)
    df["mom_10"] = close.pct_change(10)
    df["mom_20"] = close.pct_change(20)
    df["roc_3"] = close.pct_change(3)

    # --- Candlestick features ---
    df["body"] = (close - open_) / atr_14
    df["upper_wick"] = (high - pd.concat([close, open_], axis=1).max(axis=1)) / atr_14
    df["lower_wick"] = (pd.concat([close, open_], axis=1).min(axis=1) - low) / atr_14
    df["bar_range"] = (high - low) / atr_14

    # --- Volume-proxy: price range relative momentum ---
    df["high_low_ratio"] = (high - low) / close

    # --- Stochastic Oscillator (14, 3) ---
    lowest_low = low.rolling(14).min()
    highest_high = high.rolling(14).max()
    stoch_k = 100 * (close - lowest_low) / (highest_high - lowest_low).replace(0, np.nan)
    stoch_d = stoch_k.rolling(3).mean()
    df["stoch_k"] = stoch_k
    df["stoch_d"] = stoch_d
    df["stoch_kd_diff"] = stoch_k - stoch_d

    # --- Rolling volatility ---
    df["vol_10"] = close.pct_change().rolling(10).std()
    df["vol_20"] = close.pct_change().rolling(20).std()
    df["vol_ratio"] = df["vol_10"] / df["vol_20"].replace(0, np.nan)

    # --- Lagged RSI and EMA cross (for sequential signal detection) ---
    df["rsi_14_lag1"] = rsi_14.shift(1)
    df["rsi_14_lag2"] = rsi_14.shift(2)
    df["ema_cross_lag1"] = df["ema_cross"].shift(1)
    df["ema_cross_lag2"] = df["ema_cross"].shift(2)

    # --- Trend alignment: both EMAs agree ---
    df["trend_aligned_bull"] = np.where((ema_9 > ema_21) & (ema_21 > ema_50), 1, 0)
    df["trend_aligned_bear"] = np.where((ema_9 < ema_21) & (ema_21 < ema_50), 1, 0)

    # --- RSI momentum divergence proxy ---
    price_chg_5 = close.pct_change(5)
    rsi_chg_5 = rsi_14.diff(5)
    df["rsi_price_div"] = np.where(
        (price_chg_5 > 0) & (rsi_chg_5 < 0), -1,
        np.where((price_chg_5 < 0) & (rsi_chg_5 > 0), 1, 0)
    )

    # --- SMA 50 distance (for trend_filter compatibility) ---
    df["sma_50"] = close.rolling(50).mean()

    df = df.bfill().ffill()
    return df


# SECTION 2 — STRATEGY CONFIG
def strategy_config():
    return {
        "title": "AUD/USD EMA Cross RSI Gradient Boost Scalper",
        "model_type": "GradientBoostingClassifier",
        "model_params": {
            "n_estimators": 400,
            "max_depth": 4,
            "learning_rate": 0.04,
            "subsample": 0.8,
            "max_features": "sqrt",
            "min_samples_leaf": 20,
            "min_samples_split": 40,
            "validation_fraction": 0.1,
            "n_iter_no_change": 30,
            "tol": 1e-4,
        },
        "signal_threshold": 0.55,
        "direction": "both",
        "stop_loss": 0.005,
        "take_profit": 0.010,
        "cooldown": 0,
        "max_positions": 1,
        "on_opposite": "reverse",
        "session_filter": [0, 23],
        "min_atr": None,
        "trend_filter": None,
        "target_horizon": 4,
        "objective": (
            "Maximize risk-adjusted return (Sharpe) on AUD/USD 15-min bars. "
            "GradientBoostingClassifier with shrinkage (lr=0.04), moderate depth (4), "
            "subsampling (0.8) and sqrt feature fraction controls overfitting on a noisy FX "
            "series. Early stopping (n_iter_no_change=30) prevents over-training. "
            "SL=0.5%/TP=1.0% gives 1:2 RR. Threshold=0.55 filters low-confidence signals. "
            "EMA 9/21 crossover with RSI 14 confirmation is the primary signal logic, "
            "reinforced by MACD, Bollinger Bands, Stochastic, and multi-period momentum."
        ),
        "notes": (
            "Features: EMA 9, 21, 50, 200 distances; RSI 14 with overbought/oversold flags; "
            "MACD histogram; Bollinger %B and width; Stochastic K/D; ATR-normalized candle "
            "body/wicks; 5/10/20-bar momentum; rolling volatility ratio; trend alignment flags; "
            "RSI-price divergence proxy. Target horizon 4 bars (1 hour ahead). "
            "All features are lagged or rolling — no lookahead bias."
        ),
    }

# ╔══════════════════════════════════════════════════════════════╗
# ║                  STRATEGY REQUEST LOG                       ║
# ╚══════════════════════════════════════════════════════════════╝
# Generated        : 2026-05-06 02:24:20
# Model            : Gradient Boosting
# Feature Eng.     : BB (20,2.0), RSI 14, Stochastic (14,3) + Auto-add features: ON
# Signal / Entry   : Enter when model confidence > threshold; exit on opposite signal or SL/TP
# Optimization     : Maximize risk-adjusted return
# Risk Mgmt        : Stop loss 0.5%, Take profit 1.0%
# Risk Filter      : —
# ══════════════════════════════════════════════════════════════

# ============================================================
# SECTION 0 — IMPORTS & CONSTANTS
import numpy as np
import pandas as pd

DATA_PATH = "/root/Desktop/QuantifyMe/data/ohlc/AUDUSD_15min.parquet"
START_DATE = "2025-04-24"
END_DATE   = "2026-04-24"
VALIDATION_DATE = ""
TRAIN_SPLIT = 0.7

# SECTION 1 — FEATURE ENGINEERING
def feature_engineering(df, close, open_, high, low):
    # ── Bollinger Bands (20, 2) ──────────────────────────────────────────────
    bb_period = 20
    bb_std    = 2.0
    bb_mid    = close.rolling(bb_period).mean()
    bb_std_   = close.rolling(bb_period).std(ddof=1)
    bb_upper  = bb_mid + bb_std * bb_std_
    bb_lower  = bb_mid - bb_std * bb_std_
    df["bb_mid"]   = bb_mid
    df["bb_upper"] = bb_upper
    df["bb_lower"] = bb_lower
    df["bb_width"] = (bb_upper - bb_lower) / bb_mid
    df["bb_pct"]   = (close - bb_lower) / (bb_upper - bb_lower)

    # ── RSI (14) ─────────────────────────────────────────────────────────────
    rsi_period = 14
    delta      = close.diff()
    gain       = delta.clip(lower=0)
    loss       = -delta.clip(upper=0)
    avg_gain   = gain.ewm(com=rsi_period - 1, min_periods=rsi_period).mean()
    avg_loss   = loss.ewm(com=rsi_period - 1, min_periods=rsi_period).mean()
    rs         = avg_gain / avg_loss.replace(0, np.nan)
    df["rsi"]  = 100.0 - (100.0 / (1.0 + rs))

    # ── Stochastic Oscillator (K=14, D=3) ────────────────────────────────────
    stoch_k = 14
    stoch_d = 3
    low_min  = low.rolling(stoch_k).min()
    high_max = high.rolling(stoch_k).max()
    k_raw    = 100.0 * (close - low_min) / (high_max - low_min).replace(0, np.nan)
    df["stoch_k"] = k_raw
    df["stoch_d"] = k_raw.rolling(stoch_d).mean()

    # ── ATR (14) ─────────────────────────────────────────────────────────────
    atr_period = 14
    prev_close = close.shift(1)
    tr         = pd.concat([
        high - low,
        (high - prev_close).abs(),
        (low  - prev_close).abs()
    ], axis=1).max(axis=1)
    df["atr"]  = tr.ewm(com=atr_period - 1, min_periods=atr_period).mean()
    df["natr"] = df["atr"] / close

    # ── Trend / Momentum features ─────────────────────────────────────────────
    df["sma_20"]  = close.rolling(20).mean()
    df["sma_50"]  = close.rolling(50).mean()
    df["sma_100"] = close.rolling(100).mean()

    df["price_vs_sma20"]  = (close - df["sma_20"])  / df["sma_20"]
    df["price_vs_sma50"]  = (close - df["sma_50"])  / df["sma_50"]
    df["price_vs_sma100"] = (close - df["sma_100"]) / df["sma_100"]
    df["sma20_vs_sma50"]  = (df["sma_20"] - df["sma_50"]) / df["sma_50"]

    # ── MACD (12, 26, 9) ─────────────────────────────────────────────────────
    ema12        = close.ewm(span=12, min_periods=12).mean()
    ema26        = close.ewm(span=26, min_periods=26).mean()
    macd_line    = ema12 - ema26
    macd_signal  = macd_line.ewm(span=9, min_periods=9).mean()
    df["macd"]         = macd_line
    df["macd_signal"]  = macd_signal
    df["macd_hist"]    = macd_line - macd_signal

    # ── Rate-of-Change features ───────────────────────────────────────────────
    for p in [4, 8, 16]:
        df[f"roc_{p}"] = close.pct_change(p)

    # ── Volatility regime ────────────────────────────────────────────────────
    df["vol_8"]  = close.pct_change().rolling(8).std()
    df["vol_20"] = close.pct_change().rolling(20).std()
    df["vol_ratio"] = df["vol_8"] / df["vol_20"].replace(0, np.nan)

    # ── Candle body / shadow features ────────────────────────────────────────
    df["body"]        = (close - open_).abs()
    df["upper_shadow"] = high - pd.concat([close, open_], axis=1).max(axis=1)
    df["lower_shadow"] = pd.concat([close, open_], axis=1).min(axis=1) - low
    df["body_ratio"]  = df["body"] / (high - low).replace(0, np.nan)

    # ── RSI-derived features ──────────────────────────────────────────────────
    df["rsi_above_50"] = np.where(df["rsi"] > 50, 1, 0)
    df["rsi_overbought"] = np.where(df["rsi"] > 70, 1, 0)
    df["rsi_oversold"]   = np.where(df["rsi"] < 30, 1, 0)
    df["rsi_lag1"] = df["rsi"].shift(1)
    df["rsi_lag4"] = df["rsi"].shift(4)

    # ── Stochastic-derived features ───────────────────────────────────────────
    df["stoch_cross_up"]   = np.where((df["stoch_k"] > df["stoch_d"]) &
                                      (df["stoch_k"].shift(1) <= df["stoch_d"].shift(1)), 1, 0)
    df["stoch_cross_down"] = np.where((df["stoch_k"] < df["stoch_d"]) &
                                      (df["stoch_k"].shift(1) >= df["stoch_d"].shift(1)), 1, 0)
    df["stoch_oversold"]   = np.where(df["stoch_k"] < 20, 1, 0)
    df["stoch_overbought"] = np.where(df["stoch_k"] > 80, 1, 0)

    # ── BB-derived features ───────────────────────────────────────────────────
    df["bb_squeeze"] = np.where(df["bb_width"] < df["bb_width"].rolling(50).quantile(0.20), 1, 0)
    df["above_bb_upper"] = np.where(close > bb_upper, 1, 0)
    df["below_bb_lower"] = np.where(close < bb_lower, 1, 0)
    df["bb_pct_lag1"]    = df["bb_pct"].shift(1)
    df["bb_pct_lag4"]    = df["bb_pct"].shift(4)

    # ── Session hour (UTC) ────────────────────────────────────────────────────
    df["hour_utc"] = df.index.hour if hasattr(df.index, "hour") else 0

    # ── Fill NaN from indicator warm-up ──────────────────────────────────────
    df = df.bfill().ffill()
    return df

# SECTION 2 — STRATEGY CONFIG
def strategy_config():
    return {
        "title":            "AUD/USD Stochastic BB Mean-Reversion (GBM)",
        "model_type":       "GradientBoostingClassifier",
        "model_params":     {
            "n_estimators":      400,
            "max_depth":         4,
            "learning_rate":     0.04,
            "subsample":         0.80,
            "min_samples_leaf":  20,
            "max_features":      "sqrt",
            "n_iter_no_change":  30,
            "validation_fraction": 0.10,
            "tol":               1e-4,
        },
        "signal_threshold": 0.55,
        "direction":        "both",
        "stop_loss":        0.005,
        "take_profit":      0.010,
        "cooldown":         0,
        "max_positions":    1,
        "on_opposite":      "reverse",
        "session_filter":   None,
        "min_atr":          None,
        "trend_filter":     None,
        "target_horizon":   4,
        "objective": (
            "Maximize risk-adjusted return (Sharpe/Calmar) on AUD/USD 15-min. "
            "GradientBoostingClassifier with moderate depth and learning rate chosen "
            "to balance bias-variance. 2:1 reward-to-risk (SL=0.5%, TP=1.0%). "
            "Stochastic crossovers, BB mean-reversion, and RSI regime signals "
            "form the core feature set; MACD, volatility, and candle features add "
            "context. Early stopping (n_iter_no_change=30) prevents overfitting."
        ),
        "notes": (
            "Features: Bollinger Bands (20,2) width/pct, RSI(14) with lag/regime flags, "
            "Stochastic(14,3) K/D with crossover detection, ATR/NATR volatility, MACD "
            "histogram, short/medium SMAs, ROC(4/8/16), volatility ratio, candle body "
            "ratios, and UTC session hour. No session or trend filter to allow full "
            "mean-reversion opportunities across all sessions."
        ),
    }

# ╔══════════════════════════════════════════════════════════════╗
# ║                  STRATEGY REQUEST LOG                       ║
# ╚══════════════════════════════════════════════════════════════╝
# Generated        : 2026-05-06 01:51:31
# Model            : XGBoost
# Feature Eng.     : BB (20,2.0), RSI 14, Stochastic (14,3) + Auto-add features: ON
# Signal / Entry   : Enter when model confidence > threshold; exit on opposite signal or SL/TP
# Optimization     : Maximize risk-adjusted return
# Risk Mgmt        : Stop loss 0.5%, Take profit 1.0%
# Risk Filter      : —
# ══════════════════════════════════════════════════════════════

# ============================================================
# SECTION 0 — IMPORTS & CONSTANTS
import numpy as np
import pandas as pd

DATA_PATH = "/root/Desktop/QuantifyMe/data/ohlc/AUDUSD_15min.parquet"
START_DATE = "2025-04-24"
END_DATE   = "2026-04-24"
VALIDATION_DATE = ""
TRAIN_SPLIT = 0.7

# SECTION 1 — FEATURE ENGINEERING
def feature_engineering(df, close, open_, high, low):

    # ── Bollinger Bands (20, 2) ──────────────────────────────────────────────
    bb_period = 20
    bb_std    = 2.0
    bb_mid    = close.rolling(bb_period).mean()
    bb_std_val = close.rolling(bb_period).std(ddof=0)
    bb_upper  = bb_mid + bb_std * bb_std_val
    bb_lower  = bb_mid - bb_std * bb_std_val

    df["bb_mid"]   = bb_mid
    df["bb_upper"] = bb_upper
    df["bb_lower"] = bb_lower
    df["bb_width"] = (bb_upper - bb_lower) / bb_mid
    df["bb_pct"]   = (close - bb_lower) / (bb_upper - bb_lower)

    # ── RSI (14) ─────────────────────────────────────────────────────────────
    rsi_period = 14
    delta      = close.diff()
    gain       = delta.clip(lower=0)
    loss       = -delta.clip(upper=0)
    avg_gain   = gain.ewm(com=rsi_period - 1, min_periods=rsi_period).mean()
    avg_loss   = loss.ewm(com=rsi_period - 1, min_periods=rsi_period).mean()
    rs         = avg_gain / avg_loss.replace(0, np.nan)
    df["rsi"]  = 100 - (100 / (1 + rs))

    # ── Stochastic Oscillator (K=14, D=3) ────────────────────────────────────
    stoch_k_period = 14
    stoch_d_period = 3
    lowest_low     = low.rolling(stoch_k_period).min()
    highest_high   = high.rolling(stoch_k_period).max()
    denom          = (highest_high - lowest_low).replace(0, np.nan)
    df["stoch_k"]  = 100 * (close - lowest_low) / denom
    df["stoch_d"]  = df["stoch_k"].rolling(stoch_d_period).mean()
    df["stoch_kd_diff"] = df["stoch_k"] - df["stoch_d"]

    # ── Additional derived features ──────────────────────────────────────────

    # RSI overbought / oversold zone flags
    df["rsi_ob"]   = np.where(df["rsi"] > 70, 1, 0)
    df["rsi_os"]   = np.where(df["rsi"] < 30, 1, 0)
    df["rsi_mid"]  = df["rsi"] - 50.0

    # Stochastic overbought / oversold zone flags
    df["stoch_ob"] = np.where(df["stoch_k"] > 80, 1, 0)
    df["stoch_os"] = np.where(df["stoch_k"] < 20, 1, 0)

    # BB position regime: price relative to bands
    df["price_above_bb_upper"] = np.where(close > bb_upper, 1, 0)
    df["price_below_bb_lower"] = np.where(close < bb_lower, 1, 0)
    df["price_vs_bb_mid"]      = close - bb_mid

    # ATR-based volatility (14-bar)
    atr_period = 14
    tr = pd.concat([
        high - low,
        (high - close.shift(1)).abs(),
        (low  - close.shift(1)).abs()
    ], axis=1).max(axis=1)
    df["atr14"]  = tr.ewm(com=atr_period - 1, min_periods=atr_period).mean()
    df["natr14"] = df["atr14"] / close

    # SMA trend context
    df["sma_20"]  = close.rolling(20).mean()
    df["sma_50"]  = close.rolling(50).mean()
    df["sma_200"] = close.rolling(200).mean()
    df["price_vs_sma20"]  = (close - df["sma_20"]) / df["sma_20"]
    df["price_vs_sma50"]  = (close - df["sma_50"]) / df["sma_50"]
    df["sma20_vs_sma50"]  = (df["sma_20"] - df["sma_50"]) / df["sma_50"]

    # Momentum: rate of change
    df["roc_5"]  = close.pct_change(5)
    df["roc_10"] = close.pct_change(10)
    df["roc_20"] = close.pct_change(20)

    # MACD-style (EMA 12 - EMA 26)
    ema12        = close.ewm(span=12, min_periods=12).mean()
    ema26        = close.ewm(span=26, min_periods=26).mean()
    macd_line    = ema12 - ema26
    macd_signal  = macd_line.ewm(span=9, min_periods=9).mean()
    df["macd"]        = macd_line
    df["macd_signal"] = macd_signal
    df["macd_hist"]   = macd_line - macd_signal

    # Candle body / wick ratios
    body          = (close - open_).abs()
    candle_range  = (high - low).replace(0, np.nan)
    df["body_ratio"]  = body / candle_range
    df["upper_wick"]  = (high - pd.concat([close, open_], axis=1).max(axis=1)) / candle_range
    df["lower_wick"]  = (pd.concat([close, open_], axis=1).min(axis=1) - low) / candle_range
    df["bullish_bar"]  = np.where(close > open_, 1, 0)

    # Lagged RSI / Stoch features (1 and 2 bars back)
    df["rsi_lag1"]     = df["rsi"].shift(1)
    df["rsi_lag2"]     = df["rsi"].shift(2)
    df["stoch_k_lag1"] = df["stoch_k"].shift(1)
    df["bb_pct_lag1"]  = df["bb_pct"].shift(1)

    # RSI slope
    df["rsi_slope"] = df["rsi"] - df["rsi"].shift(3)

    # Stoch K crossing D (momentum signal)
    df["stoch_cross_up"]   = np.where((df["stoch_k"] > df["stoch_d"]) &
                                       (df["stoch_k"].shift(1) <= df["stoch_d"].shift(1)), 1, 0)
    df["stoch_cross_down"] = np.where((df["stoch_k"] < df["stoch_d"]) &
                                       (df["stoch_k"].shift(1) >= df["stoch_d"].shift(1)), 1, 0)

    # Volume (if present)
    if "volume" in df.columns:
        vol_ma = df["volume"].rolling(20).mean()
        df["vol_ratio"] = df["volume"] / vol_ma.replace(0, np.nan)
    else:
        df["vol_ratio"] = 1.0

    # ── Fill NaN from warm-up ────────────────────────────────────────────────
    df = df.bfill().ffill()

    return df

# SECTION 2 — STRATEGY CONFIG
def strategy_config():
    return {
        "title":            "AUD/USD Stoch+BB+RSI Mean-Reversion XGBoost",
        "model_type":       "XGBClassifier",
        "model_params": {
            "n_estimators":       400,
            "max_depth":          4,
            "learning_rate":      0.04,
            "subsample":          0.75,
            "colsample_bytree":   0.70,
            "min_child_weight":   5,
            "gamma":              0.15,
            "reg_alpha":          0.10,
            "reg_lambda":         1.50,
            "objective":          "binary:logistic",
            "random_state":       42,
            "n_jobs":             -1,
        },
        "signal_threshold": 0.55,
        "direction":        "both",
        "stop_loss":        0.005,
        "take_profit":      0.010,
        "cooldown":         0,
        "max_positions":    1,
        "on_opposite":      "reverse",
        "session_filter":   [0, 23],
        "min_atr":          None,
        "trend_filter":     None,
        "target_horizon":   4,
        "objective": (
            "Maximize risk-adjusted return (Sharpe / Calmar). "
            "XGBoost chosen for its ability to capture non-linear interactions "
            "between Stochastic, Bollinger Bands, and RSI regimes. "
            "Shallow trees (max_depth=4) + high regularisation (reg_lambda=1.5, gamma=0.15) "
            "prevent overfitting on 15-min FX data. "
            "2:1 TP:SL ratio (1.0% / 0.5%) improves expectancy per trade. "
            "Reverse on opposite signal minimises flat time and captures regime flips."
        ),
        "notes": (
            "Features include BB width/pct, RSI(14) with overbought/oversold flags, "
            "Stochastic K/D crossovers, MACD histogram, ATR volatility, SMA trend context, "
            "candle body ratios, lagged indicators, and momentum ROC. "
            "signal_threshold=0.55 balances precision vs recall on directional calls. "
            "session_filter covers full 24h to capture Asia + London + NY sessions for AUD/USD."
        ),
    }