Zhouyang626/Kronos-V4-8stocks-5min

Kronos-V4 8Stocks 5min Joint Model

基于 NeoQuasar/Kronos-small 微调的 A 股多标的联合预测模型。

模型信息

项目	详情
基座模型	NeoQuasar/Kronos-small
微调数据	8 只 A 股 × 6 年 5 分钟 K 线
总数据量	461,856 条
训练方式	分块接力续训 (4-Part Sequential Chunking)
训练设备	CPU (纯 CPU 训练)
训练耗时	~60 小时

训练标的

股票	代码
捷佳伟创	sz.300724
帝尔激光	sz.300776
迈为股份	sz.300751
西藏城投	sh.600773
富临精工	sz.300432
德业股份	sh.605117
艾罗能源	sh.688717
拉普拉斯	sh.688726

使用方法

from kronos import Kronos, KronosTokenizer, KronosPredictor

tokenizer = KronosTokenizer.from_pretrained("Zhouyang626/Kronos-V4-8stocks-5min", subfolder="tokenizer/best_model")
model = Kronos.from_pretrained("Zhouyang626/Kronos-V4-8stocks-5min", subfolder="basemodel/best_model")
predictor = KronosPredictor(model, tokenizer, device="cpu", max_context=512)

适用场景

• 最佳: 5 分钟 K 线日内短线预测 (25 分钟 ~ 4 小时)
• 可用: 同类型 A 股中小创科技股的 5 分钟预测
• 不适合: 日线及以上级别的预测

训练配置

• Lookback window: 64
• Predict window: 5
• Max context: 512
• Batch size: 4 (accumulation_steps: 4)
• Learning rate: 8e-6 (predictor), 3e-4 (tokenizer)