Kronos 金融K线预测模型上手安装、第一次预测与常见坑【免费下载链接】KronosKronos: A Foundation Model for the Language of Financial Markets项目地址: https://gitcode.com/GitHub_Trending/kronos14/Kronos面对来自 45 家全球交易所的海量 K 线靠人眼找规律基本不现实。Kronos 是第一个专为金融 K 线训练的开源基础模型给它一段历史 K 线它能接着预测未来的 K 线适合想快速体验金融预测的开发者。 它能帮你做什么输入历史 OHLCV 数据输出未来任意根数 K 线的开高低收、成交量预测一次加载模型后可连续预测多只标的支持predict_batch批量并行推理概率式采样预测温度、top-p、多路采样平均结果不是单条宿命曲线自带 A 股微调与回测流水线可以验证预测信号在策略里是否真的有效它是怎么工作的30秒版Kronos 的核心是两阶段框架先把连续的 K 线数据量化成一串离散 token相当于把行情翻译成模型能读的语言再让一个 decoder-only 的自回归 Transformer 在这些 token 上预训练像语言模型逐词生成一样逐根 K 线地写出未来。类比来说它把看盘这件事变成了接着往下造句而且是在 45 家交易所的数据上练出来的。从零跑通环境、安装与第一次调用环境要求Python 3.10依赖通过 requirements.txt 安装含 PyTorch 2.0有 GPU 更好CPU 也能跑但慢首次运行需联网从 Hugging Face 下载预训练权重git clone https://gitcode.com/GitHub_Trending/kronos14/Kronos cd Kronos pip install -r requirements.txt最短示例假设 df、x_timestamp、y_timestamp 已按 K 线数据准备好from model import Kronos, KronosTokenizer, KronosPredictor tokenizer KronosTokenizer.from_pretrained(NeoQuasar/Kronos-Tokenizer-base) model Kronos.from_pretrained(NeoQuasar/Kronos-small) predictor KronosPredictor(model, tokenizer, max_context512) pred_df predictor.predict(df, x_timestamp, y_timestamp, pred_len120) print(pred_df.head())输出的pred_df是一张 DataFrame按你给定的未来时间戳逐行列出预测的开、高、低、收、成交量和成交额拿到手就可以直接画图或喂给策略。一个真实场景走一遍预测 600977 的 5 分钟 K 线输入一份带timestamps列和 open/high/low/close 字段的 5 分钟 K 线 CSVvolume、amount 可缺省。操作仓库里现成的 examples/prediction_example.py 就是完整演示——取前 400 根 K 线作历史上下文让模型往后推 120 根跑完自动画图。看到的结果下图上半部分是收盘价蓝线为真实行情、红线为 Kronos 的预测走势与真实值明显贴合下半部分同理对比成交量。如果想看得更全面examples/yuce/ 下有多只股票的四联分析图价格、成交量、变化率、市场因素评分如果想在策略层面验证信号finetune 目录提供了 A 股微调 回测流水线下图即回测得到的累计收益曲线⚠️ 几个容易踩的坑lookback 超过 512Kronos-small / base 的最大上下文是 512 根超出的部分会被自动截断别指望更长的历史窗口列名不齐或缺失值DataFrame 必须含 open/high/low/closevolume/amount 缺失会填零但出现 NaN 会直接报错先清洗数据批量预测约束predict_batch要求所有序列的历史长度、预测长度完全一致混着不同窗口长度会失败首次下载慢权重来自 Hugging Face网络不畅时配置镜像代理再跑CPU 跑得慢设备自动检测CUDA MPS CPU没有 GPU 建议先用 4.1M 参数的 Kronos-mini 试水跑通 example 之后基本就算入门了。想看批量预测或 CSV 微调的更多细节去下方链接找脚本。相关资源完整预测示例含绘图examples/prediction_example.py无成交量数据的预测示例examples/prediction_wo_vol_example.py批量预测示例examples/prediction_batch_example.pyA 股微调与回测流水线finetune/基于 CSV 的微调说明含港股 5 分钟 K 线示例finetune_csv/README_CN.md【免费下载链接】KronosKronos: A Foundation Model for the Language of Financial Markets项目地址: https://gitcode.com/GitHub_Trending/kronos14/Kronos创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考