GitHub - shiyu-coder/Kronos: Kronos: A Foundation Model for the Language of Financial Markets
2 days ago
- #Time Series Forecasting
- #Financial AI
- #Open Source
- Kronos 是一个开源的金融K线(蜡烛图)基础模型,基于全球超过45家交易所的数据进行训练。
- 它采用两阶段框架:分词器将OHLCV数据量化为分层离散标记,然后自回归Transformer在这些标记上进行预训练。
- 该模型设计用于多样化的量化任务,支持预测和微调,并提供BTC/USDT预测的实时演示。
- 预训练模型(迷你版、小型版、基础版、大型版)已在Hugging Face Hub发布,具有不同的容量和上下文长度。
- 预测功能通过KronosPredictor类实现,涵盖预处理、归一化和预测,包括对多个时间序列的批量预测。
- 微调流程使用Qlib进行数据准备和回测,包含分词器和预测器的适配步骤。
- 项目包含数据预处理、训练和评估的脚本,并以中国A股市场数据为例进行说明。
- 提供研究引用和许可(MIT许可证),并声明部分代码注释由AI生成。