Initial commit: cc-cursor 全链路量化研究平台

7 Sprints 全部完成:
  Sprint 0: 基础设施 (DataManager + MariaDB)
  Sprint 1: 因子引擎 (34因子/12分类)
  Sprint 2: VectorBT 回测 (5策略+截面)
  Sprint 3: Optuna 优化 (+Walk-Forward)
  Sprint 4: ML 模型 (LightGBM+CatBoost)
  Sprint 5: Qwen 情绪因子 (三源新闻+日期对齐)
  Sprint 6: Agent 系统 (4Agent+日报.md/.html)

生产加固 (15项): Tushare双源fallback, SSH自动恢复, pool_pre_ping,
  save_daily先删后插, load_dotenv绝对路径, 日报5d/20d修复,
  RiskAgent改上证指数, 昨日对比+数据截止, mac_report utf8mb4,
  CLAUDE-*.md 9条已知Bug, demo全参数化, djapi数据源归一化,
  indexDatas API修正

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
2026-06-07 15:59:05 +08:00
co-authored by Claude Opus 4.7
commit 271a9343a5
293 changed files with 59598 additions and 0 deletions
@@ -0,0 +1,51 @@
"""
因子阈值交叉策略。
通用策略:任意因子上穿/下穿阈值 → 交易信号。
支持:
- 上穿买入 (cross_up: close < MA → cross above MA → buy)
- 下穿买入 (cross_down: RSI > 70 → cross below 30 → buy)
"""
import pandas as pd
from backtest.base import BaseStrategy
from backtest.signal import factor_to_threshold_signal
class FactorCrossStrategy(BaseStrategy):
"""
因子阈值交叉策略。
适用场景:
- 均线偏离度上穿 0 → 买入(趋势转多)
- 波动率下穿阈值 → 买入(波动收敛后突破)
"""
category = "trend"
def __init__(
self,
factor_column: str,
buy_threshold: float, # 因子大于此值买
sell_threshold: float | None = None,
cross_direction: str = "up",
):
self.factor_column = factor_column
self.buy_threshold = buy_threshold
self.sell_threshold = sell_threshold
self.cross_direction = cross_direction
self.name = f"factor_cross_{factor_column}"
def generate_signals(self, factor_df: pd.DataFrame) -> pd.Series:
if self.factor_column not in factor_df.columns:
raise ValueError(f"factor_df 缺少 '{self.factor_column}'")
factor = factor_df[self.factor_column]
return factor_to_threshold_signal(
factor,
buy_threshold=self.buy_threshold,
sell_threshold=self.sell_threshold,
cross_direction=self.cross_direction,
)
@@ -0,0 +1,58 @@
"""
因子排序轮动策略。
定期按因子值排序,买入排名最高的股票(截面策略)。
"""
import pandas as pd
from backtest.base import BaseStrategy
class FactorRotationStrategy(BaseStrategy):
"""
因子排序选股策略。
适用于多股票截面场景:对每只股票计算因子值,
选排名最高的 top_n 只做多。
"""
category = "rotation"
def __init__(self, factor_name: str, top_n: int = 5, bottom_n: int = 0):
self.factor_name = factor_name
self.top_n = top_n
self.bottom_n = bottom_n
self.name = f"rotation_{factor_name}_top{top_n}"
def generate_signals(self, factor_df: pd.DataFrame) -> pd.Series:
"""
单股票/截面模式:factor_df 支持两种输入方式。
- 单股票: 对每只股票单次调用
- 截面: 通过 run_cross_section 逐股票调用
"""
if self.factor_name not in factor_df.columns:
raise ValueError(f"factor_df 缺少 '{self.factor_name}'")
factor = factor_df[self.factor_name]
valid = factor.dropna()
if len(valid) < self.top_n * 2:
return pd.Series(-1, index=factor_df.index)
threshold = valid.quantile(1 - self.top_n / max(len(valid), self.top_n))
signals = pd.Series(-1, index=factor_df.index)
signals[factor > threshold] = 1
return signals
def rank_stocks(
self, factor_values: dict[str, float]
) -> list[str]:
"""
对股票按因子值排序,返回 top N 的 ts_code 列表。
参数:
factor_values: {ts_code: factor_value}
"""
sorted_stocks = sorted(factor_values, key=factor_values.get, reverse=True)
return sorted_stocks[: self.top_n]
@@ -0,0 +1,52 @@
"""
动量突破策略。
价格突破 N 日新高 → 买入
价格跌破 N 日均线 → 平仓
"""
import pandas as pd
from backtest.base import BaseStrategy
from backtest.signal import factor_to_threshold_signal
class MomentumBreakoutStrategy(BaseStrategy):
"""动量突破策略。"""
category = "trend"
def __init__(self, lookback: int = 20, exit_period: int = 10):
self.lookback = lookback
self.exit_period = exit_period
self.name = f"mom_breakout_{lookback}"
def generate_signals(self, factor_df: pd.DataFrame) -> pd.Series:
if "close" not in factor_df.columns:
raise ValueError("factor_df 缺少 'close'")
close = factor_df["close"]
# 买入信号:突破 N 日新高
rolling_high = close.rolling(window=self.lookback, min_periods=self.lookback).max()
breakout = close >= rolling_high.shift(1)
# 平仓信号:跌破 exit 日均线
exit_ma = close.rolling(window=self.exit_period, min_periods=self.exit_period).mean()
signals = pd.Series(0, index=close.index)
signals[breakout] = 1
signals[close < exit_ma] = 0
return self._dedup(signals)
@staticmethod
def _dedup(signals: pd.Series) -> pd.Series:
"""只保留第一个买入和第一个卖出信号。"""
result = signals.copy()
prev = -1
for i in range(len(result)):
if result.iloc[i] == prev:
result.iloc[i] = -1
else:
prev = result.iloc[i]
return result[result != -1].reindex(signals.index).fillna(-1)
@@ -0,0 +1,35 @@
"""
RSI 均值回归策略。
RSI 低于超卖线 → 买入
RSI 高于超买线 → 平仓
"""
import pandas as pd
from backtest.base import BaseStrategy
from backtest.signal import factor_to_threshold_signal
class RSIMeanRevertStrategy(BaseStrategy):
"""RSI 超买超卖反转策略。"""
category = "mean_revert"
def __init__(self, oversold: float = 30, overbought: float = 70, rsi_column: str = "rsi_14"):
self.oversold = oversold
self.overbought = overbought
self.rsi_column = rsi_column
self.name = f"rsi_revert_{int(oversold)}_{int(overbought)}"
def generate_signals(self, factor_df: pd.DataFrame) -> pd.Series:
if self.rsi_column not in factor_df.columns:
raise ValueError(f"factor_df 缺少 '{self.rsi_column}'")
rsi = factor_df[self.rsi_column]
return factor_to_threshold_signal(
rsi,
buy_threshold=self.oversold,
sell_threshold=self.overbought,
cross_direction="down", # RSI 向下跌破 oversold → 买入
)
+33
View File
@@ -0,0 +1,33 @@
"""
均线交叉策略。
短期均线上穿长期均线 → 买入
短期均线下穿长期均线 → 平仓
"""
import pandas as pd
from backtest.base import BaseStrategy
from backtest.signal import cross_signal
class SMACrossStrategy(BaseStrategy):
"""快慢均线交叉策略。"""
category = "trend"
def __init__(self, fast: int = 5, slow: int = 20):
self.fast = fast
self.slow = slow
self.name = f"sma_cross_{fast}_{slow}"
def generate_signals(self, factor_df: pd.DataFrame) -> pd.Series:
if "close" not in factor_df.columns:
raise ValueError("factor_df 缺少 'close'")
close = factor_df["close"]
min_p = min(self.fast, self.slow)
ma_fast = close.rolling(window=self.fast, min_periods=self.fast).mean()
ma_slow = close.rolling(window=self.slow, min_periods=self.slow).mean()
return cross_signal(ma_fast, ma_slow)