feat(training): 添加数据过滤器支持及 ST 股票过滤

- 新增 filters.py 模块,实现 BaseFilter 抽象类和 STFilter 过滤器
- 在 Trainer 中支持 filters 参数,可在股票池筛选之前执行数据过滤
- 更新 training/__init__.py 导出 BaseFilter 和 STFilter
- 在 regression.py 中集成 STFilter,用于过滤 ST 股票
This commit is contained in:
2026-03-04 21:14:39 +08:00
parent f1687dadf3
commit af5c96cd53
4 changed files with 168 additions and 1 deletions

View File

@@ -33,6 +33,9 @@ from src.training.components.processors import (
# 模型
from src.training.components.models import LightGBMModel
# 数据过滤器
from src.training.components.filters import BaseFilter, STFilter
# 训练核心
from src.training.core import StockPoolManager, Trainer
@@ -57,6 +60,9 @@ __all__ = [
"StandardScaler",
"CrossSectionalStandardScaler",
"Winsorizer",
# 数据过滤器
"BaseFilter",
"STFilter",
# 模型
"LightGBMModel",
# 训练核心