三步出图:Kronos K线预测模型完整上手指南
【免费下载链接】KronosKronos: A Foundation Model for the Language of Financial Markets项目地址: https://gitcode.com/GitHub_Trending/kronos14/Kronos
Kronos 是一个开源的金融K线预测基础模型,把K线行情"翻译"成符号序列后,像预测下一个词一样推断未来走势。它解决的痛点是:普通时间序列模型很难同时消化开、高、低、收、量多个维度还夹着大量噪声的K线数据。无论你是想快速看图的散户,还是有自己行情的开发者,都能零门槛拿到预测结果。
它是谁:一句话说清 Kronos 的定位
- 首个面向金融K线的开源基础模型,论文已被 AAAI 2026 接收
- 预训练数据覆盖 45 家以上全球交易所的行情,不绑定单一市场
- 面向三类读者:想看效果的散户、有自己K线CSV的开发者、要做微调的量化研究者
定位讲清了,下面先跑起来再说别的。
Kronos快速上手:十分钟拿到第一个预测结果
先装环境,Python 3.10+,然后装依赖:
git clone https://gitcode.com/GitHub_Trending/kronos14/Kronos cd Kronos pip install -r requirements.txt最省事的路径是网页界面,一条命令启动:
cd webui python run.py浏览器打开localhost:7070后,操作流很顺:选 CSV 数据文件 → 选模型和设备(CPU/CUDA/MPS)→ 拖时间窗口(固定 400 个历史点加 120 个预测点)→ 点预测,图就出来了。数据格式只要求open、high、low、close四列,volume、amount可选;首次加载模型会从 Hugging Face 拉权重,稍等一下。想写代码的可以直接调KronosPredictor,传入历史数据、历史时间戳和要预测的未来时间戳即可,完整可运行脚本见 examples/prediction_example.py。
跑通之后,最有价值的就是看懂它的输出。
看懂预测图:模型真实输出长这样
第一张是官方示例脚本的产物:左边是历史K线,右边接上模型预测出来的走势,可以直接比对"延续"的方向合不合理。
第二张是微调后的回测曲线:实线是 top-K 简单策略的累计收益,虚线是基准,一眼看出这段时间策略是否跑赢市场。
第三张是真实个股平安银行 000021 的预测图,examples/yuce/目录里还配了 JSON 分析报告,能翻到单只股票输出的细节。
图看懂了,下一个问题自然冒出来:它凭什么能预测。
拆掉黑箱:Kronos 凭什么预测K线
核心实现在 model/,思路一句话就能说清:语言模型靠"分词"读文章,K线也能分词。第一步是 tokenizer(分词器),把开高低收量这类连续多维数据量化成两层粗细不同的离散符号——行话叫层级化 token,白话点说,一层抓大体走势、一层保细节。第二步是自回归 Transformer,在符号序列上做预训练,学会"根据前面发生了什么,推断下一步更可能发生了什么"。它不是去拟合一条曲线,而是学走势的分布,这正好对上高噪声的金融数据。
机制讲完了,轮到挑模型了。
按显卡和需求选模型规格
- Kronos-mini:4.1M 参数,上下文长度 2048,轻量,适合快速试跑
- Kronos-small:24.7M 参数,上下文长度 512,速度和精度平衡,README 示例默认用它
- Kronos-base:102.3M 参数,上下文长度 512,精度更高
- Kronos-large:499.2M 参数,未开源
💡 实用提醒:small 和 base 的上下文上限是 512,喂进去的历史窗口(lookback)别超过这个数,超了KronosPredictor会自动截断。建议先用 mini 把数据格式和流程跑通,再换大模型。
规格定下来之后,如果你有自己的数据,就可以往深里走了。
Kronos微调指南:批量预测和进阶训练入口
- 有自己CSV的开发者→ finetune_csv/:改一份 yaml 配置就能训练,内置阿里巴巴港股 09988 五分钟K线的完整示例,配置参考
configs/config_ali09988_candle-5min.yaml - 量化研究者→ finetune/:Qlib 数据处理、tokenizer 与 predictor 两阶段微调(支持 torchrun 多卡)、回测评估一整套脚本链,可作为自建策略的起点
- 要批量跑多只股票→ 调
predict_batch方法,多个标的并行预测,一次拿到全部结果
最后说点不粉饰的。
避坑与边界:官方免责说了什么
- ⚠️ Qlib 回测流水线官方明确说是演示,不是生产级量化交易系统
- 模型输出属于"原始预测"信号,离实盘还差组合优化、风险因子中性化、交易成本建模一整套工程
- 回测曲线好看,也不等于闭眼买入的依据,风险管理得自己上
Kronos 把"读懂K线"这件难事,收敛成了分词加自回归预测一条清晰链路,模型、数据和代码都摆在仓库里。对普通人,它是零门槛的预测可视化入口;对研究者,它是一个能持续微调、回测验证的基座。
【免费下载链接】KronosKronos: A Foundation Model for the Language of Financial Markets项目地址: https://gitcode.com/GitHub_Trending/kronos14/Kronos
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考