feat(training): 添加 LightGBM LambdaRank 排序学习功能
新增基于 LambdaRank 的排序学习模型,用于股票排序预测任务: - 实现 LightGBMLambdaRankModel 模型类,支持分位数标签转换 - 提供完整的训练流程和 NDCG 评估指标 - 添加实验 Notebook 演示排序学习全流程
This commit is contained in:
@@ -4,5 +4,6 @@
|
||||
"""
|
||||
|
||||
from src.training.components.models.lightgbm import LightGBMModel
|
||||
from src.training.components.models.lightgbm_lambdarank import LightGBMLambdaRankModel
|
||||
|
||||
__all__ = ["LightGBMModel"]
|
||||
__all__ = ["LightGBMModel", "LightGBMLambdaRankModel"]
|
||||
|
||||
Reference in New Issue
Block a user