ARTICLE DETAIL

资讯详情

深耕郑州网站建设与运营推广的一线实战洞察。

Kronos 金融 K 线预测基础模型完整指南:如何 4 步跑通第一个 K 线预测

Kronos 金融 K 线预测基础模型完整指南:如何 4 步跑通第一个 K 线预测 Kronos 金融 K 线预测基础模型完整指南如何 4 步跑通第一个 K 线预测【免费下载链接】KronosKronos: A Foundation Model for the Language of Financial Markets项目地址: https://gitcode.com/GitHub_Trending/kronos14/Kronos预测 K 线这件事过去要么手工堆砌技术指标要么从零训练一个时间序列模型成本高、周期长、还容易过拟合。Kronos 把这条路打通了它是首个开源的金融 K 线蜡烛图基础模型在 45 个以上全球交易所的 K 线数据上预训练完成装好依赖后几行代码就能对任意品种的 OHLCV 序列做未来预测。这篇指南带你从零环境搭建、首次运行一路走到参数调优和避坑完整走一遍。一、先搞懂Kronos 到底解决了什么问题传统做法里你每换一个市场、换一只股票、换一种频率几乎都要重新建模。而金融数据本身噪声极大、信噪比低从零训模型更是难上加难。Kronos 的思路和 GPT 很神似GPT 先读完整个互联网的语言之后翻译、总结、写代码都不需要单独训练Kronos 则先把 45 交易所的 K 线序列读完学会了一套通用的市场语言——价格怎么走、波动怎么延续。这样你拿到它之后直接预测喂一段历史 K 线它就能续写未来几根 K 线微调适配在你自己的数据比如某只港股的 5 分钟线上继续训练让它更懂你的标的。用一句话概括它不是又一套需要从头调的指标公式而是一位已经考完大学的预测员你只需要告诉它看哪段行情、预测多长。二、快速上手从零到跑出第一个 K 线预测4 步✅ 全程不需要写模型代码下面每一步都有现成脚本。第 1 步克隆仓库并安装依赖⚡️环境要求 Python 3.10git clone https://gitcode.com/GitHub_Trending/kronos14/Kronos cd Kronos pip install -r requirements.txt依赖很轻PyTorch、pandas、matplotlib 加上 Hugging Face 相关库见 requirements.txt。第 2 步准备一份历史 K 线数据一个 CSV 即可必须包含open / high / low / close四列volume / amount可选再加一列timestamps。数据频率不限日线、分钟线都行。第 3 步跑官方预测示例python examples/prediction_example.py脚本会自动完成加载 Kronos-small 与配套分词器首次运行从官方模型库下载权重→ 取 400 根历史 K 线作上下文 → 预测未来 120 根 → 画出真实值 vs 预测值的收盘价与成交量对比图。想看没有成交量数据的预测可以换 examples/prediction_wo_vol_example.py。第 4 步可选用图形界面操作不想碰代码的话仓库内置了一个 Flask 版 Web UI支持 CSV/Feather 数据、K 线图展示和参数调节cd webui python run.py浏览器访问http://localhost:7070选数据、选模型mini / small / base、拖时间窗口、点预测即可详见 webui/README.md。三、核心能力拆解Kronos 是怎么做到的整个体系可以理解成压缩 → 学习 → 续写三级流水线核心有三块能力。1️⃣ K 线分词器把连续行情压成词表K 线的开高低收量本来就是连续数值没法直接当词用。Kronos-Tokenizer 先对一段 K 线做粗粒度量化再补一层细粒度的残差量化源码里的s1、s2两级码本见 model/module.py。打个比方这就像视频压缩——先存一帧关键帧大致形状再逐帧存和上一帧差多少细节修正。两级码本配合既保留了 K 线的大形态也不丢微观结构。2️⃣ 自回归 Transformer像背课文一样学下一根 K 线拿到离散 token 后一个 decoder-only 的因果 Transformer 在海量 K 线语料上做下一 token 预测预训练。它和 GPT 逐词续写的逻辑完全一致只是词汇换成了 K 线 token。关键在于因果掩码预测第 N 根 K 线时只能看到前 N-1 根未来信息物理上无法泄漏——这正是回测和实盘最需要的那种干净。3️⃣ KronosPredictor一个入口搞定全套推理model/ 里的KronosPredictor把归一化、分词、采样、反归一化全部封装进predict()你只传三个东西历史 DataFrame、历史时间戳、要预测的未来时间戳。而且它和 LLM 一样带采样旋钮T温度控制预测的随机程度top_p核采样控制考虑多少种可能的未来sample_count生成多条预测路径再平均降低单次抽样的抖动。四、真实使用场景把它用起来场景 1单只标的的短期行情预测拿 400 根 5 分钟 K 线当上下文预测接下来 120 根5 小时。examples/prediction_example.py跑完就能得到这样的对比图预测曲线红与实际走势蓝的重合度一目了然场景 2多资产组合的批量预测要同时盯一篮子股票或期货别写循环——用predict_batch。把多个 DataFrame 和对应时间戳打包传入GPU 并行推理每个序列独立做归一化。注意两个硬约束所有序列的历史长度必须一致预测长度pred_len也必须一致。这对每日收盘后批量刷新全组合预测这类任务非常合适。场景 3用自己的数据微调 回测验证预训练模型是通才微调之后才成专才。仓库提供两条路线CSV 快速路线推荐新手准备好含 OHLCV 的 CSV改一下 YAML 里的路径和窗口参数然后cd finetune_csv python train_sequential.py --config configs/config_ali09988_candle-5min.yaml脚本会按先 tokenizer、后 predictor的顺序自动训完多卡可用torchrun起 DDP。仓库里附了阿里港股 5 分钟线的完整示例数据finetune_csv/data/微调后的预测效果如下Qlib 完整路线面向 A 股日线流程是配置finetune/config.py→ 数据切分qlib_data_preprocess.py→torchrun微调 tokenizer 和 predictor → 跑qlib_test.py生成 top-K 策略回测输出策略对基准的累积收益曲线五、进阶与避坑从能用到好用的调优清单 下面几条都是实际跑起来最容易踩的点。上下文长度是硬上限Kronos-small / base 的max_context是 512mini 是 2048。上下文再长也会被自动截断所以lookback建议别超过 512否则前面那段历史根本没用上。模型选型看显存mini4.1M 参数跑得飞快适合大批量扫数据small24.7M是速度和质量的平衡点官方示例默认用它base102.3M质量最高但显存吃紧。CPU、CUDA、MPS 都支持M 系 Mac 直接可用。采样参数怎么调默认T1.0, top_p0.9适合快速验证想要更稳的预测路径可参考 Web UI 的推荐值——T 调到 1.2~1.5、top_p 0.95~1.0、sample_count设 2~3 取平均多次采样能明显抹平单次抽样的毛刺。数据格式的坑批量预测时长度不一致是最常见的报错来源缺少volume / amount列会自动补零但open / high / low / close一个都不能少。从 Demo 到实盘的提醒⚠️模型输出的是原始预测信号官方也明确说回测示例只是演示、不是生产级交易系统。真要落地至少还要做组合优化、风险因子中性化并在回测里认真建模交易成本、滑点和冲击成本——这一步省了回测曲线会骗你。结语Kronos 把K 线预测从每个标的一个模型变成了一个预训练基础模型 轻量微调几行代码就能出预测几分钟就能起一个微调实验。现在就去把它跑起来克隆仓库、装好依赖、执行python examples/prediction_example.py今天就能看到属于你自己的第一条预测曲线。【免费下载链接】KronosKronos: A Foundation Model for the Language of Financial Markets项目地址: https://gitcode.com/GitHub_Trending/kronos14/Kronos创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
返回列表