INNER CODE UNIT · Python

build_equity_chart_and_data

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

def build_equity_chart_and_data(ticker: str, params: dict, train_ratio: float = 0.7):
    data = MarketDataSkill(ticker).execute()
    ta   = TechnicalAnalysisSkill(data).add_indicators().dropna().copy()

    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")

View source record →