INNER CODE UNIT · Python

split_idx

alex-jb/orallexa-ai-trading-agent · app.py:187

    split_idx = int(len(ta) * train_ratio)
    train_df  = ta.iloc[:split_idx].copy()
    test_df   = ta.iloc[split_idx:].copy()

    bt_train = simple_backtest(train_df, params=params, debug=False)
    bt_test  = simple_backtest(test_df,  params=params, debug=False)
    return bt_train, bt_test


def render_equity_chart(ticker: str, bt_train: pd.DataFrame, bt_test: pd.DataFrame):
    fig, axes = plt.subplots(2, 1, figsize=(10, 7))

    axes[0].plot(bt_train["CumulativeStrategyReturn"], label="Train Strategy")
    axes[0].plot(bt_train["CumulativeMarketReturn"],   label="Train Buy & Hold")
    axes[0].set_title(f"{ticker} Train Equity Curve")
    axes[0].legend()
    axes[0].grid(True)

View source record →