liaozhaorun
5a1f278df8
- 将因子定义、模型参数、日期配置提取为模块级常量 - 优化 LightGBM 参数（降低过拟合风险） - LightGBMModel 支持 params 字典参数传入 - 修复 StockFilter 创业板排除逻辑（支持 301xxx） - 添加 experiment/output 到 .gitignore
82 lines
2.4 KiB
Python
82 lines
2.4 KiB
Python
"""股票池选择器配置
|
||
|
||
提供股票过滤和市值选择的配置类。
|
||
"""
|
||
|
||
from dataclasses import dataclass
|
||
from typing import List, Optional
|
||
|
||
|
||
@dataclass
|
||
class StockFilterConfig:
|
||
"""股票过滤器配置
|
||
|
||
用于过滤掉不需要的股票（如创业板、科创板等）。
|
||
基于股票代码进行过滤，不依赖外部数据。
|
||
|
||
Attributes:
|
||
exclude_cyb: 是否排除创业板（300xxx, 301xxx）
|
||
exclude_kcb: 是否排除科创板（688xxx）
|
||
exclude_bj: 是否排除北交所（.BJ 后缀）
|
||
exclude_st: 是否排除ST股票（需要外部数据支持）
|
||
"""
|
||
|
||
exclude_cyb: bool = True
|
||
exclude_kcb: bool = True
|
||
exclude_bj: bool = True
|
||
exclude_st: bool = True
|
||
|
||
def filter_codes(self, codes: List[str]) -> List[str]:
|
||
"""应用过滤条件，返回过滤后的股票代码列表
|
||
|
||
Args:
|
||
codes: 原始股票代码列表
|
||
|
||
Returns:
|
||
过滤后的股票代码列表
|
||
|
||
Note:
|
||
ST 股票过滤需要额外数据，在 StockPoolManager 中处理。
|
||
此方法仅基于代码前缀进行过滤。
|
||
"""
|
||
result = []
|
||
for code in codes:
|
||
# 排除创业板（300xxx, 301xxx）
|
||
if self.exclude_cyb and code.startswith(("300", "301")):
|
||
continue
|
||
# 排除科创板（688xxx）
|
||
if self.exclude_kcb and code.startswith("688"):
|
||
continue
|
||
# 排除北交所（.BJ 后缀）
|
||
if self.exclude_bj and code.endswith(".BJ"):
|
||
continue
|
||
result.append(code)
|
||
return result
|
||
|
||
|
||
@dataclass
|
||
class MarketCapSelectorConfig:
|
||
"""市值选择器配置
|
||
|
||
每日独立选择市值最大或最小的 n 只股票。
|
||
市值数据从 daily_basic 表独立获取，仅用于筛选。
|
||
|
||
Attributes:
|
||
enabled: 是否启用选择
|
||
n: 选择前 n 只
|
||
ascending: False=最大市值, True=最小市值
|
||
market_cap_col: 市值列名（来自 daily_basic）
|
||
"""
|
||
|
||
enabled: bool = True
|
||
n: int = 100
|
||
ascending: bool = False
|
||
market_cap_col: str = "total_mv"
|
||
|
||
def __post_init__(self):
|
||
"""验证配置参数"""
|
||
if self.n <= 0:
|
||
raise ValueError(f"n 必须是正整数，得到: {self.n}")
|
||
if not self.market_cap_col:
|
||
raise ValueError("market_cap_col 不能为空")
|