Kronos K线预测实战指南:四步读懂首个开源金融基础模型
【免费下载链接】KronosKronos: A Foundation Model for the Language of Financial Markets项目地址: https://gitcode.com/GitHub_Trending/kronos14/Kronos
Kronos 是首个面向金融 K 线(蜡烛图)序列的开源基础模型,在全球 45+ 交易所数据上预训练,几行代码即可预测股票未来的 K 线与成交量。
📈 先看效果:预测的K线和真实K线有多接近
先看结果。下图来自官方示例 examples/prediction_example.py:模型用最近 400 个历史数据点,预测出未来 120 个数据点。
Kronos K 线预测示例:预测的收盘价与成交量曲线(红色)对照真实数据(蓝色)
- 涨跌大方向跟随真实曲线,关键转折点不缺席
- 预测的成交量峰谷与真实量能基本对齐
- 一次预测同时给出开、高、低、收、量五维完整 K 线
这意味着什么?你不用分别做价格预测和成交量预测,一次调用就拿到一条完整 K 线,直接叠加到行情软件里就能看。
🎯 Kronos解决了什么问题
如果你是看技术分析的散户,或者是刚入门的量化开发者,可能被下面这些事卡住过:
- 数据维度多、噪音大:K 线有开、高、低、收、量五个维度,还掺杂突发消息和情绪,均线、MACD 这类指标很难抓住拐点
- 一个模型只能伺候一只股票:传统机器学习流水线,换一只标的、换一个市场、换一个周期,基本都要重新训练,成本很高
- 通用时序模型不适配 K 线:通用时间序列基础模型没有为金融数据的高噪音特性专门设计
- 自己训练基础模型不现实:预训练数据、Token 化方案、算力资源,对普通开发者都是门槛
这意味着什么?市面上缺一个"拿来就能用"的预训练 K 线模型,而 Kronos 恰好填上了这个空。
🔍 核心原理通俗版:K线是如何变成"语言"的
Kronos 的架构是两阶段设计,可以类比"写文章":
Kronos 系统架构:K 线数据经 Tokenizer 离散化后,由自回归 Transformer 完成预测
- 第一步:Token 化。Tokenizer 把连续的多维 K 线数据(OHLCV)量化成分层的离散 Token,就像把 K 线图"翻译"成模型能读的句子:先按周期切出一个个"字",每个"字"里再精细记录价格与量能的细节
- 第二步:自回归预测。因果 Transformer 块堆叠成语言模型,在 Token 序列上训练。模型预测"下一个字"时只能看历史、不能偷看未来——和真实交易场景完全一致
模型家族提供四个规格:mini(410 万参数)、small(2470 万)、base(1.02 亿)均已开源,上下文长度最高 2048(mini 版),核心源码在 model/ 目录。
这意味着什么?Kronos 把 K 线预测当成"语言建模"来做,你只管"说同一种语言",它就能给你续写出下一段 K 线。
🚀 快速上手:四步跑通第一次K线预测
克隆仓库并安装环境
需要 Python 3.10+:
git clone https://gitcode.com/GitHub_Trending/kronos14/Kronos cd Kronos pip install -r requirements.txt准备K线CSV数据
标准 CSV 即可:必须包含timestamps、open、high、low、close列,volume和amount可选。仓库里的 finetune_csv/data/ 有一份现成样例,格式照着看。
注意:small/base 模型最大上下文为 512 个数据点,输入更长时KronosPredictor会自动截断。
运行首次预测并查看结果
跑一下官方示例脚本即可,它自动完成加载模型、准备数据、预测和绘图:
python examples/prediction_example.pypredict方法返回一个 pandas DataFrame,按你给的时间戳索引,包含预测的开高低收、成交量与成交额,还可以用T(温度)、top_p、sample_count控制概率采样;要一次预测多只股票,用predict_batch批量并行即可。完整调用链只有KronosTokenizer+Kronos+KronosPredictor三个类,更多变体可翻 examples/ 目录。
这意味着什么?即使没写过深度学习代码,只要会读一个 CSV,十几分钟就能拿到第一条预测曲线。
🧪 进阶玩法:WebUI与自定义数据微调
零代码体验:WebUI
webui/ 目录内置了一个本地 Web 界面,启动后访问http://localhost:7070:
cd webui python run.py流程全在浏览器里完成:选数据文件 → 选模型与设备(CPU/CUDA/MPS 都支持)→ 调温度、采样参数 → 用滑杆选定 400+120 个数据点的时间窗 → 一键预测,自动出 K 线图并和真实数据做误差对比。
用自己的数据微调模型
仓库提供两条微调管线,对应两类数据:
- A 股日线:finetune/ 目录基于 Qlib 准备数据,流程是"配置 → 数据划分 → 微调 Tokenizer → 微调 Predictor → 回测",回测会输出策略累计收益曲线对照基准:
微调后的模型跑通回测:策略累计收益与超额收益曲线示例
- 任意 CSV 数据:finetune_csv/ 目录更简单,一条命令顺序训练 Tokenizer 和 Predictor:
python train_sequential.py --config configs/config_ali09988_candle-5min.yaml。官方示例用阿里巴巴 5 分钟 K 线微调后的效果如下:
Kronos 5 分钟 K 线微调示例:在阿里巴巴(09988)高频数据上微调后的预测效果
这意味着什么?A 股、港股、加密货币的分钟级数据,只要有 CSV,你都能把 Kronos 微调成贴着自己市场走的版本。
💎 为什么选Kronos
- 开源且学术背书:MIT 协议免费商用,论文已被 AAAI 2026 接收
- 低配机器也能跑:mini 版仅 410 万参数,WebUI 支持 CPU 和 Mac 的 MPS,不必须高端显卡
- 生态开箱即用:单股预测、批量预测、无成交量预测、微调、回测都有现成脚本,端到端能跑通
- 可定制不封闭:Tokenizer 与 Predictor 两个组件可独立微调,换成自己的数据分布就能继续用
Kronos 把 K 线预测从"每只股票训一个模型"变成了"加载预训练基础模型、接上你的数据就能跑"。现在就动手吧:先按上面步骤克隆仓库装好依赖,跑通 examples/prediction_example.py 拿到第一条预测曲线,再挑一只你熟悉的股票,用 finetune_csv/ 微调出专属于你的模型。
实用提示:官方已说明微调示例是简化演示,并非生产级量化交易系统;预测信号用于实盘前,建议叠加组合优化、风控与交易成本建模,谨慎对待。
【免费下载链接】KronosKronos: A Foundation Model for the Language of Financial Markets项目地址: https://gitcode.com/GitHub_Trending/kronos14/Kronos
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考