BAIZE AI RESOURCE

金融K线预测基础模型

Kronos

Kronos 是首个开源的金融K线基础模型,基于全球45个交易所的数据训练,能根据历史行情预测未来价格走势,并提供网页界面和微调脚本,帮助用户完成量化预测与分析。

Kronos

资源介绍

Kronos 是面向量化研究和金融数据分析的开源 K 线预测工具。用户准备好历史行情数据后,可以调用预训练模型获得未来走势的预测结果,并通过网页界面查看 K 线图、调整预测设置和比较结果。研究人员还可以用自己的 CSV 数据微调模型,测试它在特定市场或时间尺度上的表现。它适合做行情研究、模型对比和策略探索;预测只是基于历史数据的估计,应结合回测与风险判断,不宜作为单独的交易依据。

核心能力

多规模预训练模型

提供 Kronos-mini、Kronos-small、Kronos-base 等不同参数量的模型,用户可按算力和精度需求选择,模型可从 Hugging Face 直接加载。

网页可视化预测

内置 Web UI,可选择数据文件、模型和计算设备,通过滑块选取时间窗口,一键生成预测结果并以专业K线图展示。

预测质量参数调节

支持调整温度、核采样和采样次数等参数,控制预测的随机性与多样性,帮助用户获得更符合需求的预测结果。

预测与实际对比分析

系统自动将预测结果与实际数据对比,给出价差统计、误差分析和预测质量评估,方便判断预测可信度。

自定义数据微调

提供微调脚本,支持用符合格式要求的CSV行情数据训练分词器和预测器,让模型适配用户自己的市场和标的。

多设备加速支持

支持 CPU、NVIDIA CUDA 和 Apple MPS 计算设备,用户可根据硬件条件选择运行方式。

使用方法

  1. 安装 Python 3.10 及以上版本,并安装项目依赖。

  2. 从 Hugging Face 加载预训练模型和对应的分词器。

  3. 使用 KronosPredictor 传入历史行情数据,即可得到未来价格预测。

  4. 如需图形化操作,进入 webui 目录启动服务,浏览器访问本地端口。

  5. 在网页界面中选择数据文件、模型、计算设备和预测参数,点击预测查看结果。

  6. 如需适配自有数据,按格式准备CSV文件并运行微调脚本训练模型。

使用案例

A股个股行情预测

使用场景量化研究者想预测某只A股未来走势

准备包含开高低收和成交量的历史数据,加载 Kronos-small 模型,运行预测脚本得到未来若干时间点的价格预测,并生成对比图表。

港股5分钟K线微调

使用场景用户希望模型适配港股高频数据

按格式整理港股5分钟K线CSV文件,修改配置中的数据路径和训练参数,运行顺序训练脚本完成分词器和预测器微调。

网页端快速体验

使用场景非技术用户想直观查看预测效果

启动 Web UI 后选择示例数据文件和模型,调整温度与采样参数,点击预测即可在K线图上查看预测结果和误差分析。

包含内容

kronos-source.zip
README-中文安装说明.md
LICENSE-MIT.txt
SHA256.txt

常见问题

Kronos 网页界面支持导入哪些数据格式?
网页界面支持 CSV、Feather 等金融数据格式。
Kronos 网页界面的预测时间窗口可以调整吗?
网页界面使用固定的 400 个历史数据点加 120 个预测数据点的时间窗口,可通过滑块选择窗口位置。
Kronos 网页界面支持哪些计算设备?
网页界面支持选择 CPU、CUDA 或 MPS 设备运行预测。
Kronos 的预测结果可以调整吗?
网页界面可调整温度、核采样参数和采样次数等设置;其中温度影响预测的随机性,核采样参数影响结果的多样性。
用自己的数据微调 Kronos 需要准备什么?
需要准备包含 timestamps、open、high、low、close、volume、amount 列的 CSV 文件;没有成交量或成交额时,对应列可填 0。还需在训练配置中设置数据路径和预训练模型路径。