拓冰建站拓冰建站
首页 / 资讯中心 / 正文

Kronos K线价格预测完整指南:从数据清洗到回测的5个步骤

Kronos K线价格预测完整指南从数据清洗到回测的5个步骤【免费下载链接】KronosKronos: A Foundation Model for the Language of Financial Markets项目地址: https://gitcode.com/GitHub_Trending/kronos14/KronosKronos 是首个开源的K线金融基础模型基于45家全球交易所的数据预训练专门用于预测K线序列OHLCV开盘、最高、最低、收盘、成交量、成交额。它采用两阶段架构先用 tokenizer 把连续的多维K线数据量化成分层离散 token再用自回归 Transformer 在这些 token 上学习走势规律。这篇文章讲清 Kronos 金融价格预测的完整流程数据准备 → 模型选型 → 推理配置 → 结果验证 → 微调迭代每步都给出具体数字和对应配置文件的位置。第一步数据准备——喂给模型前的5项检查Kronos 对输入格式的硬性要求很简单必备列timestamps、open、high、low、close可选列volume、amount没有就填 0时间戳连续、无缺口频率统一比如全程5分钟K线标准样例在 finetune_csv/data/HK_ali_09988_kline_5min_all.csv格式规范见 finetune_csv/README_CN.md。拿到数据后做5项检查列名是否符合规范、时间戳是否连续、单点价格跳变离群值是否处理、频率是否一致、序列长度是否超过模型上下文限制。最后一条尤其关键——官方预测脚本会自动做归一化但不会帮你修数据脏数据会直接传导进预测结果。第二步模型选型——4个规格一次看懂Kronos 家族按参数量分4档全部托管在 Hugging Face模型参数量上下文长度是否开源Kronos-mini4.1M2048是Kronos-small24.7M512是Kronos-base102.3M512是Kronos-large499.2M512否选型结论很直接CPU 或单卡快速验证mini上下文最长2048日常预测任务small官方回测和示例脚本都基于它显存充足、追求精度baselarge 未开源目前拿不到注意 small 和 base 的max_context都是 512这是模型能处理的最大序列长度直接决定下一步的窗口配置。第三步推理配置——T、top_p、sample_count 三个旋钮加载模型只需两行KronosPredictor会自动处理归一化、推理、反归一化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)调用predict时有三个参数值得调pred_df predictor.predict( dfx_df, x_timestampx_ts, y_timestampy_ts, pred_len120, T1.2, # 温度控制生成随机性越大走势越多样 top_p0.95, # 核采样只在累计概率≥0.95的token里抽取 sample_count4 # 生成4条路径取平均降低单条预测的方差 )经验区间T取 1.0~1.5官方 Web UI 推荐 1.2~1.5top_p取 0.9~1.0sample_count取 3~5。想要更保守的输出就压低温度官方回测脚本用的就是T0.6, top_p0.9, sample_count5见 finetune/config.py 末尾的inference_*配置。历史窗口怎么分512 的上下文要在历史和预测之间分配历史占 75%~80%预测长度不超过 25%。官方示例的切法是lookback400 pred_len120Web UI 也固定了这个窗口见 webui/。超过 512 的输入会被自动截断但主动控制长度更可控。批量预测的正确姿势多资产同时预测用predict_batch它走 GPU 并行、逐序列独立归一化。约束有三条缺一条就跑不起来所有序列的历史长度lookback相同所有序列的pred_len相同每个 DataFrame 都含open/high/low/close可运行的脚本见 examples/prediction_batch_example.py它对同一条5分钟数据切5个窗口并行预测。第四步结果验证——别只信一条预测验证分三层从便宜到严格1. 目视检查。预测的收盘价不应偏离历史太多high和low应包住open/close。跑 examples/prediction_example.py 会生成真值 vs 预测的对照图2. 策略回测。微调管线内置了回测入口python finetune/qlib_test.py --device cuda:0它对测试集做批量推理生成预测信号后跑 Top-K 策略对照基准如沪深300配置在 finetune/config.py 的_set_benchmark。累计收益曲线示例官方提醒这套回测是简化演示不是生产级量化系统真实策略还需要组合优化、风险因子中性化、交易成本和滑点建模。3. 回归测试。tests/test_kronos_regression.py 用固定随机种子 top_k1跑确定性推理和预存的期望输出比对256/512 两种上下文。换机器、换依赖后先跑它能快速确认环境没被装坏。第五步微调——把 Kronos 适配到你的市场预训练覆盖的是全球市场通用规律落到某个市场或某只票上微调收益明显。仓库提供两条管线Qlib 管线finetune/以A股为例改 finetune/config.py 里的路径和超参默认lookback_window90, predict_window10, epochs30, batch_size50归一化裁剪clip5.0python finetune/qlib_data_preprocess.py切分训练/验证/测试集先用torchrun跑finetune/train_tokenizer.py再跑finetune/train_predictor.py用finetune/qlib_test.py回测验证学习率也有讲究tokenizer 用 2e-4predictor 用 4e-5前者新训练成分多后者更接近继续预训练。CSV 管线finetune_csv/不依赖 Qlib一个 CSV 一份 YAML 配置就能跑train_sequential.py一次完成 tokenizer 和 predictor 的顺序训练支持 DDP 多卡。完整说明见 finetune_csv/README_CN.md。在阿里巴巴099885分钟K线上微调后的预测效果让模型持续变好微调不是终点。建议每季度重新回测一次对比策略收益是否衰减市场风格变化时优先重调T和sample_count成本最低其次才考虑用新数据重训。速查清单数据timestamps OHLC 必备volume/amount可填 0上下文small/base 上限 512窗口建议 400120采样T1.0~1.5top_p0.9~1.0sample_count3~5批量预测lookback 和 pred_len 必须全序列统一验证顺序目视 → 回测 → 回归测试微调先 tokenizer 后 predictor顺序不能反【免费下载链接】KronosKronos: A Foundation Model for the Language of Financial Markets项目地址: https://gitcode.com/GitHub_Trending/kronos14/Kronos创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
分享:

看完干货,该让你的企业上线了

免费需求沟通 · 48 小时内出具建站方案 · 河南本地可上门