社区所有版块导航
Python
python开源   Django   Python   DjangoApp   pycharm  
DATA
docker   Elasticsearch  
aigc
aigc   chatgpt  
WEB开发
linux   MongoDB   Redis   DATABASE   NGINX   其他Web框架   web工具   zookeeper   tornado   NoSql   Bootstrap   js   peewee   Git   bottle   IE   MQ   Jquery  
机器学习
机器学习算法  
Python88.com
反馈   公告   社区推广  
产品
短视频  
印度
印度  
Py学习  »  Git

GitHub 3.8万 Star,清华团队做了一个读 A 股 K 线的大模型!

Java基基 • 2 月前 • 335 次点击  

👉 这是一个或许对你有用的社群

🐱 一对一交流/面试小册/简历优化/求职解惑,欢迎加入「芋道快速开发平台」知识星球。下面是星球提供的部分资料: 

👉这是一个或许对你有用的开源项目

国产Star破10w的开源项目,前端包括管理后台、微信小程序,后端支持单体、微服务架构

RBAC权限、数据权限、SaaS多租户、商城、支付、工作流、大屏报表、ERP、CRM、WMS、MES、HRM、AI大模型、IoT物联网、IM聊天等功能:

  • 多模块:https://gitee.com/zhijiantianya/ruoyi-vue-pro
  • 微服务:https://gitee.com/zhijiantianya/yudao-cloud
  • 视频教程:https://doc.iocoder.cn
【国内首批】支持 JDK25 + SpringBoot4、JDK17/21+SpringBoot3、JDK8/11+Spring Boot2三版本 

本期导读

•

P0|先别急着开终端,网页上就能白嫖一张预测图

•

P1|三行命令进门,官方示例数据就是上交所的 A 股

•

P2|选尺寸就是选车位,4.1M 起步,各有各的停法

•

P3|三个旋钮,把预测从抽奖拧成参考

•

P4|扫全市场,predict_batch 的两条等长规矩

•

P5|微调自己的票,四步流水线加一张显存账单

•

P6|台阶尽头的牌子,演示不是生产

会写 Python、对 A 股好奇的人,多半动过「让 AI 帮我看 K 线」的念头。K 线不神秘,红一根绿一根,背后就是开盘、收盘、最高、最低、成交量几个数。

最近很多人在 GitHub 刷到同一个项目,教大模型读这些数字,读完接着往后预测,Star 冲到 3.7 万,Fork 六千出头。

项目叫 Kronos,清华团队做的,论文刚被 AAAI 2026 收录。同类示例大多是美股或海外数据,它等于替国内开发者把 A 股这条路先走了一遍。

IMG-001
IMG-001

它是全球第一个面向金融 K 线的开源基础模型,出自清华大学交叉信息研究院和自动化系。思路一句话,大语言模型读文字,它读 K 线,开高低收和成交量转成离散 token,用 Transformer 训练。

训练底子不小,全球 45 家交易所的 120 多亿条 K 线记录,股票、期货、数字货币都有,覆盖 7 种时间粒度。

Kronos 项目定位卡,清华两院出品,45 家交易所 120 亿根 K 线训练
Kronos 项目定位卡,清华两院出品,45 家交易所 120 亿根 K 线训练

01

PART

P0|先别急着开终端,网页上就能白嫖一张预测图

KRONOS FIELD NOTE

第一步不是装环境,打开 shiyu-coder.github.io/Kronos-demo 就能看 BTC/USDT 未来 24 小时预测,一行代码不用写。

想动手的人有多少?B 站讲实测微调的视频播放 1.5 万左右、点赞两百多,收藏却近千。收藏是点赞三倍多,标准「马住回头跑」。

02

PART

P1|三行命令进门,官方示例数据就是上交所的 A 股

KRONOS FIELD NOTE

想本地跑门槛也低,Python 3.10 以上就行。先把仓库拉下来装依赖。

装完就能上手,核心就这几行。

返回的 pred_df 是预测出的开高低收和成交量,能画图、算指标。examples/prediction_example.py 从读数据到画图全写好,直接执行就出图。

最顺的在路径里,XSHG_5min_600977.csv 是官方示例,上交所 5 分钟 K 线,从这步起打交道的就是 A 股,不用折腾数据源,社区实测一块消费级显卡就能玩。

没有成交量也别慌,仓库现成 prediction_wo_vol_example.py,volume 和 amount 两列不给,自动补零。

有个坑,输入历史长度别超模型上下文,small 和 base 都是 512,超了自动截断。示例 lookback=400 已贴天花板,想喂更长历史看下一节。

IMG-003
IMG-003

官方示例里,蓝真实红预测,子图是收盘价和成交量。

IMG-004
IMG-004

03

PART

P2|选尺寸就是选车位,4.1M 起步,各有各的停法

KRONOS FIELD NOTE

四个尺寸像车位,第四种只在图纸上。

对号入座。

•

mini 是 410 万参数,笔记本跑得动,独享 2048 上下文;

•

small 是官方示例默认款,最省心;

•

base 主打精度,下载量百万级;

•

large 没放出来。

前三个在 Hugging Face 的 NeoQuasar 下随下随用,最受欢迎的分词器下载超两百万次;small/base 配 Tokenizer-base、mini 配 Tokenizer-2k,最小的模型反而看得最远。

四个尺寸选型对比,mini 4.1M 独享 2048 上下文,large 499.2M 未放出
四个尺寸选型对比,mini 4.1M 独享 2048 上下文,large 499.2M 未放出

04

PART

P3|三个旋钮,把预测从抽奖拧成参考

KRONOS FIELD NOTE

预测是概率式的,不是拍死一个数。predict 的三个参数就是旋钮。

•

T 是采样温度,越高越放飞,越低越保守;

•

top_p 是采样范围,控制从多大的候选池里抽;

•

sample_count 是采样路径数量,开大点就多跑几条路径再取平均,结果稳一些。

社区踩过坑,sample_count 太低曲线过度平滑,像趋势线,其实丢了波动信息。

sample_count 同时还是个成本旋钮,采样显存约等于资产数×预测长度×sample_count,批量跑前先算这笔账。

05

PART

P4|扫全市场,predict_batch 的两条等长规矩

KRONOS FIELD NOTE

做量化很少只盯一只票。predict_batch 把多个 DataFrame 打包传进去,GPU 并行一次出一堆结果,扫全市场省时间。

两条硬规矩,所有序列的历史窗口要一样长,预测长度也要一样长;缺成交量的列会自动补零。

predict_batch 批量预测流程,多个 DataFrame 打包,GPU 并行输出
predict_batch 批量预测流程,多个 DataFrame 打包,GPU 并行输出

06

PART

P5|微调自己的票,四步流水线加一张显存账单

KRONOS FIELD NOTE

预训练学的是公开市场的通用规律,想更懂 A 股就得微调。2025 年 8 月官方放出完整微调脚本,配了一套用 Qlib 处理 A 股数据的示例流程。

流水线四步,改配置、处理数据、先微调 tokenizer 再微调主模型、最后跑回测。

微调四步流程,改配置、Qlib 数据、微调 tokenizer、微调主模型,最后回测
微调四步流程,改配置、Qlib 数据、微调 tokenizer、微调主模型,最后回测

实操重心在第一步,finetune/config.py 改五类路径,qlib_data_path、dataset_path、save_path、backtest_result_path、预训练模型路径;不用 Comet.ml 就设 use_comet=False。

配好按顺序跑,NUM_GPUS 换成卡数。

回测直接画出策略和基准的累计收益对比图,README 演示里策略线明显压着基准走;支持 torchrun 多卡。

IMG-007
IMG-007

官方把丑话说在前面,这套流程只是演示,不是能直接上生产的量化系统,真实策略还隔着组合优化和风险中性化。

彩蛋在 finetune 里,大量注释由 Gemini 2.5 Pro 生成,官方自己提醒可能有误、以代码为准。

07

PART

P6|台阶尽头的牌子,演示不是生产

KRONOS FIELD NOTE

普通人离金融大模型的距离,从来不是一个模型的距离,是一级一级没被拆好的台阶的距离。Kronos 最值钱的地方,是把台阶从网页 Demo 修到了微调自己的票。

冷静算账,base 才 1 亿参数,跟动辄几十亿的大语言模型比是小不点。但 K 线转 token 的思路,加上权重、微调脚本、回测流程全放了出来,你拿到的不是论文,是一套能跑的东西。

调研泼了冷水,Kronos 更合理的定位是价格行为和成交结构建模模块,不是完整的投资决策系统,模型输出的是信号,不是收益。

分层建议,会写 Python、对量化好奇的,现在就玩,从 Demo 和示例数据起步;想「输入代码输出买卖点」的,先观望。不做交易,学这套思路也开眼界。以上纯属技术交流,不构成任何投资建议。

三条核心观点,思路有意思、整条链路开源、概率参考有边界
三条核心观点,思路有意思、整条链路开源、概率参考有边界

CLOSING

它适合做价格行为与成交结构建模,不是完整的投资决策系统。模型给的是概率信号,不是收益承诺。

芋芋道源码

如果你觉得今天这篇有收获,欢迎点赞、在看、转发三连,我们下篇见。

THANKS FOR READING




欢迎加入我的知识星球,全面提升技术能力。

👉 加入方式,“长按”或“扫描”下方二维码噢:

星球的内容包括:项目实战、面试招聘、源码解析、学习路线。

文章有帮助的话,在看,转发吧。

谢谢支持哟 (*^__^*)

Python社区是高质量的Python/Django开发社区
本文地址:http://www.python88.com/topic/200145