Commit Graph
62 Commits
Author SHA1 Message Date
ShiYu 1abb87ee7c Merge pull request #182 from phaoer/feature/cn-markets
Add examples from China’s A-share market
2025-11-05 17:48:37 +08:00
ShiYu ab5ef4b5ce Merge pull request #174 from AnMakc/master
Improve inference throughput
2025-11-05 17:48:13 +08:00
phaoer a46b680413 feat: China A-share market 2025-11-04 17:51:54 +08:00
Maxim b62f780de2 Refactor auto_regressive_inference to reduce memory allocations and cpu-gpy syncs. 2025-10-29 17:37:05 -03:00
ShiYu 788bc6f760 Merge pull request #173 from AnMakc/regression_test
Add regression tests
2025-10-29 23:20:55 +08:00
Maxim 64569135b2 Remove unnecessary CUDA cache clearing - no memory leak reproducible now. 2025-10-29 12:04:43 -03:00
Maxim 21eb36afc7 Do not collect tokenizer metrics during inference 2025-10-29 12:04:43 -03:00
Maxim dccfa764fc Use pytorch SDPA implementation 2025-10-29 12:04:43 -03:00
Maxim 94f3212f4c Parametrize ctx len for mse regression test. 2025-10-29 11:46:56 -03:00
Maxim a2d8a27e4e Add regression tests 2025-10-29 10:45:23 -03:00
ShiYu eeb3168f71 Merge pull request #167 from SoYuCry/master
fix: define missing split_token in HierarchicalEmbedding
2025-10-26 20:30:18 +08:00
YuCry a7e294cc56 fix: define missing split_token in HierarchicalEmbedding 2025-10-25 23:37:30 +08:00
ShiYu a5f5aba12d Merge pull request #152 from RahulPatel2727/master
Update README.md
2025-10-19 16:25:33 +08:00
Rahul Patel b4e24f3e1b Update README.md 2025-10-18 10:25:56 +05:30
ShiYu 082ab7ef62 Merge pull request #138 from Luciferbobo/master
add CSV-based finetuning pipeline for Kronos models
2025-10-12 17:06:30 +08:00
zhangboyu1 7f658d9672 update config & readme 2025-10-09 18:14:04 +08:00
zhangboyu1 166b4162fb add example data 2025-10-09 17:36:56 +08:00
zhangboyu1 a8df339586 update readme 2025-10-09 16:35:59 +08:00
zhangboyu1 38b5176cb7 update readme & figs 2025-10-09 16:24:35 +08:00
BOBO 2f7d3484cf Update README.md 2025-10-09 16:04:52 +08:00
BOBO a50b425863 Update README.md 2025-10-09 16:00:01 +08:00
zhangboyu1 814a5edb42 add vis figs 2025-10-09 15:53:50 +08:00
BOBO c5eeb50a99 Update README.md 2025-10-09 15:50:25 +08:00
zhangboyu1 84f74ae341 add custom data finetune 2025-10-09 15:48:39 +08:00
ShiYu 083294bd84 Merge pull request #93 from ArtificialZeng/patch-1
wrong expressions and typo
2025-09-16 10:36:13 +08:00
ShiYu 87157161d4 Bug fix 2025-09-16 10:35:14 +08:00
ShiYu ac69e16750 Bug fix 2025-09-16 10:32:03 +08:00
Dr. Artificial曾小健 c05233c065 wrong expressions and typo
wrong expressions and typo
2025-09-15 16:05:04 +08:00
ShiYu f82acd69bd Refactor qlib_test.py 2025-09-10 21:37:57 +08:00
ShiYu 764913b7d0 Refactor dataset and backtesting logic in qlib_test.py
Refactor QlibTestDataset and QlibBacktest classes for improved structure and readability. Update inference logic and main execution flow.
2025-09-10 21:36:27 +08:00
ShiYu a4c14cb094 Merge pull request #63 from dowithless/patch-1
doc: add links to translated README versions
2025-09-03 21:57:07 +08:00
neo ed9581b8f4 doc: add links to translated README versions
Added language selection links to the README for easier access to translated versions: German, Spanish, French, Japanese, Korean, Portuguese, Russian, and Chinese.
2025-09-03 17:53:03 +08:00
ShiYu 2d1a1ae809 Merge pull request #59 from pengxiao-song/dev_empty_cuda_cache
fix: add torch.cuda.empty_cache() during autoregressive inference
2025-09-02 10:36:56 +08:00
Pengxiao Song e027051b38 fix: add torch.cuda.empty_cache() during autoregressive inference
Without releasing cached GPU memory, usage will keep growing during autoregressive prediction, leading to significant memory increase or OOM. Calling torch.cuda.empty_cache() prevents this accumulation.
2025-09-02 10:26:27 +08:00
ShiYu 939986adb1 Merge pull request #53 from QuantML-C/master
add batch prediction
2025-09-01 21:27:10 +08:00
quant 38a643b761 update kronos model code 2025-09-01 21:22:27 +08:00
ShiYu 6ba4330c09 Merge pull request #54 from ehan1990/bugfix/missing-dep-safe-tensor
bugfix: add missing dependency safetensors==0.6.2
2025-09-01 19:37:28 +08:00
Edward Han 0eb1652711 add missing dependency safetensors==0.6.2 2025-09-01 00:54:30 -07:00
quant 1201848bc6 add batch prediction and corresponding example 2025-09-01 14:32:47 +08:00
quant 44e57bc761 add batch prediction and corresponding example 2025-09-01 14:30:42 +08:00
ShiYu 6a13c2c1a9 Merge pull request #25 from CharlesJ-ABu/feature/webui
feat: add Kronos Web UI
2025-08-27 17:13:38 +08:00
Charles 14863929a0 feat: Translate all Chinese content to English in webui module
- Translate all Chinese comments and strings in webui/app.py
- Translate all Chinese comments and strings in webui/run.py
- Translate all Chinese comments and strings in webui/start.sh
- Translate all Chinese content in webui/README.md
- Translate all Chinese content in webui/templates/index.html
- Add prediction results directory for analysis
- Complete internationalization of webui module
2025-08-27 15:36:47 +08:00
ShiYu 8ac972ec50 Refactor QlibDataPreprocessor for clarity and structure 2025-08-27 11:36:29 +08:00
ShiYu 7ba06fc286 Merge pull request #20 from xxsunyxx/master
Update qlib_data_preprocess.py
2025-08-27 10:41:28 +08:00
Charles 328dc4a3b0 fix: 修复相对变化率转绝对价格的累积计算逻辑,使用cumprod确保价格连续性 2025-08-26 17:22:59 +08:00
Charles 35d35f740f feat: 添加详细的调试信息,显示价格转换的每一步操作 2025-08-26 17:13:18 +08:00
Charles b4b270954b fix: 解决Kronos模型预测连续性问题,通过价格转换保持数据连续性 2025-08-26 17:02:24 +08:00
Charles ca2502757f refactor: 将prediction_results目录移动到webui目录下,优化项目结构 2025-08-26 16:49:56 +08:00
Charles 609235c077 feat: 添加预测结果保存功能,用于分析预测质量问题 2025-08-26 16:35:34 +08:00
Charles 1f394cace3 feat: 添加Kronos Web UI完整功能
- 集成Kronos模型(mini/small/base)
- 支持CPU/CUDA/MPS设备选择
- 时间窗口滑条选择器(400+120固定窗口)
- 预测质量参数控制(Temperature, Top-P, Sample Count)
- 预测vs实际数据对比分析
- 完整的Flask后端和现代化前端界面
- 支持CSV和Feather格式数据文件
- 完整的启动脚本和文档
2025-08-26 16:19:21 +08:00