shiyu-coder/Kronos

项目概览

该项目以超过 45 个全球交易所的数据进行预训练，并提供金融 K 线预测、微调及回测能力，填补了通用时序模型在金融高噪声数据上的空白。

项目类型

模型开发 · 模型运行 · 数据处理

应用场景

数据分析

部署方式

查看项目说明

License

MIT

适合谁

• 需要使用预训练基础模型处理金融 K 线序列的数据团队或研究员
• 希望在自有金融数据集上进行模型微调的 AI 工程师
• 需要对模型预测结果进行基础策略回测的量化研究者

核心能力

• 通过 KronosPredictor 类对金融蜡烛图数据（OHLCV）生成未来预测。
• 提供在用户自有数据集上对 tokenizer 和 predictor 模型进行微调的流水线。
• 包含使用简单 top-K 策略回测来评估微调模型性能的脚本。

限制与风险

• 提供的微调流水线是简化示例，并非生产就绪的量化交易系统。
• 模型生成的原始信号并非纯 alpha；在真实工作流中需要组合优化等复杂技术来实现稳定收益。
• finetune 目录中的代码注释由 AI 助手生成，可能存在不准确之处。

如何开始

• 需安装 Python 3.10+，并通过 pip install -r requirements.txt 安装依赖。
• 从 Hugging Face Hub 加载预训练的 Kronos 基础模型（Kronos-mini、Kronos-small 或 Kronos-base）及对应的 Tokenizer（Kronos-Tokenizer-base 或 Kronos-Tokenizer-2k），实例化 KronosPredictor 并调用 predict 方法运行预测。

核验依据

• README: Kronos is the **first open-source foundation model** for financial candlesticks (K-lines), > trained on data from over **45 global exchanges**.
• README: Kronos: A Foundation Model for the Language of Financial Markets
• README: Kronos is a family of decoder-only foundation models, pre-trained specifically for the "language" of financial markets—K-line sequences. Unlike general-purpose TSFMs, Kronos is de…
• README: 🚩 [2025.11.10] Kronos has been accpeted by AAAI 2026.
• README: Forecasting with Kronos is straightforward using the `KronosPredictor` class.

内容核验时间：