Yuxuan Yan
1caa63faeb
Replace the old signal/strategy/backtest modules with a decoupled
data → alpha → combo pipeline (parquet between phases, .pq extension).
Alphas:
- BaseAlpha + @register_alpha factory/plugin registry; one file per
built-in (reversal, reversal_vol, momentum); external alphas via
--alpha-module. Alphas are z-scored position weights, not predictors.
Data:
- baostock primary / akshare fallback, treated consistently.
- New --universe all (~5000 A-shares via query_all_stock, filtered).
- login-once batch downloader; empty-string OHLCV coerced to NaN.
- Month-partitioned dataset {output_dir}/{universe}/month=YYYY-MM/*.pq
with chunked durability flushes; --data-path is the dataset dir.
CLI logs at INFO by default (--log-level) so progress is visible.
Docs (README, CLAUDE.md) updated incl. pipeline diagram and roadmap
TODOs for portfolio construction / backtest / paper trading.
Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
19 lines
472 B
Python
19 lines
472 B
Python
"""Short-horizon momentum alpha."""
|
|
import pandas as pd
|
|
|
|
from pipeline.alpha.base import BaseAlpha
|
|
from pipeline.alpha.registry import register_alpha
|
|
|
|
|
|
@register_alpha
|
|
class MomentumAlpha(BaseAlpha):
|
|
"""Positive trailing return: stocks that rose score high."""
|
|
|
|
name = "momentum"
|
|
|
|
def __init__(self, lookback: int = 5):
|
|
self.lookback = lookback
|
|
|
|
def signal(self, close: pd.DataFrame) -> pd.DataFrame:
|
|
return close.pct_change(self.lookback)
|