Kronos K线预测快速上手:8分钟并行预测上千只股票,3步跑通第一次
Kronos K线预测快速上手8分钟并行预测上千只股票3步跑通第一次【免费下载链接】KronosKronos: A Foundation Model for the Language of Financial Markets项目地址: https://gitcode.com/GitHub_Trending/kronos14/Kronos你盯几百只股票靠人肉看K线根本盯不过来。Kronos 是首个专门处理K线序列的开源基础模型支持上千只股票并行做K线预测一批大批量分析耗时约 8 分钟。这篇就是完整的 Kronos 使用教程从安装到第一次预测只要 3 步后面还会讲参数速查、回测指标怎么读以及 Kronos 微调的完整流程。Kronos 是什么两句话看懂K线基础模型Kronos 是首个专门处理K线序列的开源基础模型核心是一个两阶段架构。先用专门的K线分词器把开高低收OHLCV这类多维K线数据量化成层次化的离散 token再在这些 token 上预训练一个自回归 Transformer把收盘价和成交量的预测统一交给同一个模型。三步跑通第一次K线预测结论先行克隆仓库、装依赖、跑官方示例脚本你就能拿到第一张预测图。第 1 步克隆仓库git clone https://gitcode.com/GitHub_Trending/kronos14/Kronos cd Kronos第 2 步安装依赖pip install -r requirements.txt第 3 步跑第一次预测示例脚本已经包含从加载数据到出图的完整流程python examples/prediction_example.py跑完你会看到一张真实值与预测值的对比图。如果手头数据没有成交量可以直接用 examples/prediction_wo_vol_example.py不需要 volume 和 amount 列。Kronos 参数速查只需记住 3 个你只需要看懂 3 个参数剩下的预处理、归一化、反归一化都由 KronosPredictor 自动完成参数作用说明max_context模型能处理的最大序列长度Kronos-small / Kronos-base 为 512超长输入会自动截断lookback输入窗口长度建议不超过 max_context效果更稳pred_len预测步长一次生成多少根未来K线从 1 只到 1000 只KronosPredictor 与 predict_batch 的分工单只股票交给 KronosPredictor千股并行预测交给 predict_batch。predict_batch 接收多份 DataFrame 和时间戳列表利用 GPU 并行一次完成多序列预测唯一的硬要求是所有序列历史长度相同、pred_len 相同。这正是批量分析沪深 300、中证 500 成分股、识别板块轮动、监控异常波动个股这些场景的基础。完整写法可以参考 examples/prediction_batch_example.py。Kronos 预测效果与回测指标怎么读先看预测对比图再看回测曲线。预测对比图把真实值和预测值叠在一起一眼就能看出收盘价与成交量两条线的走势拟合程度真实标的方面仓库里附带了港股 09988阿里巴巴5 分钟K线的完整预测实例见 finetune_csv/examples/。预测之后要进回测验证。回测结果重点读 3 个指标累积收益带成本看绝对表现超额收益带成本看扣掉交易成本后还剩多少优势基准指数是参照线策略曲线长期站在基准上方才算真正跑赢用自己的数据微调 Kronos4 步简表官方微调流水线分 4 步按顺序执行即可步骤做什么用什么1. 配置设定数据路径、模型保存路径、epochs 等超参finetune/config.py2. 数据准备用 Qlib 处理数据并切成训练/验证/测试集finetune/qlib_data_preprocess.py3. 模型微调先微调 Tokenizer再微调 Predictortorchrun 启动 finetune/train_tokenizer.py、finetune/train_predictor.py4. 回测验证测试集上生成预测信号并跑 top-K 策略回测finetune/qlib_test.py开始前先装 Qlibpip install pyqlib。完整的 Kronos 微调说明在 finetune/ 目录用 CSV 数据微调的另一种方案见 finetune_csv/README_CN.md。部署前先检查这组配置硬件要达到这条线才跑得顺GPU 显存 ≥40GB、多核 CPU、内存 ≥256GB。显存吃紧时动态批处理加混合精度能把显存占用再降 20%。效率方面官方数据口径是效率提升 53.8%、时间节省 67%、预测准确率超 85%。谁该用 Kronos以及下一步个人投资者先跑 examples/prediction_example.py体验单只股票的K线预测。资产管理机构用 predict_batch 做成分股批量分析辅助日常组合筛查。对冲基金用自有数据微调把模型输出接进自己的回测体系。下一步建议按顺序走看 README.md 了解全部功能和模型版本运行 examples/prediction_example.py 或 examples/prediction_batch_example.py想搭可视化预测页面看 webui/README.md。【免费下载链接】KronosKronos: A Foundation Model for the Language of Financial Markets项目地址: https://gitcode.com/GitHub_Trending/kronos14/Kronos创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考