K线到底能不能预测?用开源K线大模型Kronos做股票预测的完整指南
【免费下载链接】KronosKronos: A Foundation Model for the Language of Financial Markets项目地址: https://gitcode.com/GitHub_Trending/kronos14/Kronos
Kronos 是一个面向金融 K 线的开源基础模型,用 45 家以上全球交易所的行情数据预训练,能直接从历史 K 线推出未来的价格走向。它适合三类读者:想看看效果的普通投资者、手里有自己行情数据的开发者,以及需要批量微调与回测验证的量化研究者。
为什么通用时序模型一到K线就容易翻车
一根蜡烛同时装着开高低收四个维度,还夹着成交量、成交额,噪声又大。这种数据直接喂给通用时间序列模型,结果往往是"拟合历史很顺,预测未来就崩"。
Kronos 的思路很直白:先把 K 线"标准化"。就像棋手复盘棋谱——每一步先翻译成固定的棋谱坐标,记法统一了,高手才能看万局棋、学会"前面这么走,下一步大概率怎么走"。
三条命令跑起本地 Kronos K线预测
前提只有一个:Python 3.10 以上。然后克隆仓库并装好依赖:
git clone https://gitcode.com/GitHub_Trending/kronos14/Kronos cd Kronos && pip install -r requirements.txt最省事的路径是 webui/ 里的网页界面,启动就一条命令:
cd webui && python run.py浏览器打开http://localhost:7070,选中 CSV 数据文件(必须含 open、high、low、close 四列),选好模型和 CPU/CUDA/MPS 设备,拖一下时间窗口,点预测,就能看到历史 K 线与预测走势的叠加图和对比分析。首次加载模型会从 Hugging Face 拉取权重,稍等片刻即可。
💡 降低踩坑成本的小建议:先用Kronos-mini(仅 4.1M 参数)把数据格式和整套流程跑通,确认没问题后再换Kronos-base提升精度,避免一开始就被环境问题和精度问题搅在一起。
先看三种典型的 Kronos K线预测产出
仓库里放着三类产出,可以逐一对照:
① 单标的预测对比图。运行 examples/prediction_example.py 会生成历史走势与未来预测的叠加图:
② 微调后的回测曲线。用 finetune/ 里的 Qlib 流程在 A 股日线数据上微调,再跑 top-K 简单策略回测,输出策略相对基准的累计收益曲线:
③ 真实个股的预测输出。examples/yuce/ 里有平安银行 000021 等沪深个股的预测图和配套 JSON 分析报告,可以翻看模型对单只股票的输出细节:
拆开看Kronos预测模型是怎么搭的
整条链路就是"先量化、再生成"两阶段,核心实现都在 model/ 里:
- 层级量化 Tokenizer:把连续的多维 K 线数据压缩成粗细两层的离散符号,细节不丢
- 自回归 Transformer:在符号序列上学习"前面发生了什么,下一步更可能发生什么",天然适合高噪声的金融数据
- 模型库:mini / small / base 三个规格任选;small 和 base 最大上下文 512,历史窗口别超过这个长度,
KronosPredictor会自动截断 KronosPredictor预测入口:归一化、生成、反归一化一站包办,还支持predict_batch对多只标的并行预测
用之前先看清这几条 Kronos 使用限制
官方明说:微调加回测那套流程只是演示,模型输出属于"原始预测信号",离实盘还差组合优化、风险因子中性化、交易成本建模这一整套工程,别把回测曲线当成能直接上场的交易系统。
数据上也有边界:small / base 只吃最长 512 个历史点,选窗口时别超;volume、amount 两列可有可无,只有 OHLC 四列也能跑,缺列不用硬凑。
按角色选路线:从 webui 到 Qlib 微调
- 只想看看效果的散户:直接进 webui/,不写一行代码,界面里有 K 线图展示和预测/实际值对比
- 有自己数据的开发者:finetune_csv/ 提供 CSV 微调全流程,改一份 yaml 配置就能训练,内置阿里巴巴港股 09988 五分钟 K 线的完整示例
- 量化研究者:finetune/ 给出 Qlib 数据处理、tokenizer 与 predictor 两阶段微调(支持 torchrun 多卡)和回测评估的脚本链,可作为自建策略的起点
Kronos 把"读 K 线"这件难事收敛成量化加自回归预测一条清晰链路,每一步的模型、数据和代码都摆在开源仓库里。对新手它是零门槛的预测可视化入口,对研究者则是一个能持续微调、回测验证的基座模型。
【免费下载链接】KronosKronos: A Foundation Model for the Language of Financial Markets项目地址: https://gitcode.com/GitHub_Trending/kronos14/Kronos
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考