shen/stock-transformer
1
0
Fork 0
mirror of https://github.com/dupenf/stock-transformer.git synced 2024-11-25 16:08:34 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
股票人工智能模型，使用transformer模型，训练股票数据，预测未来一天的收盘价格。
3 Commits 1 Branch 0 Tags 14 MiB
Python 100%
main
Go to file
杜鹏飞 22a3e47c3c
Update README.md
2024-08-05 14:16:30 +08:00
datasets Add files via upload 2024-08-05 14:15:06 +08:00
__init__.py Add files via upload 2024-08-05 14:15:06 +08:00
a0_config.py Add files via upload 2024-08-05 14:15:06 +08:00
d0_download.py Add files via upload 2024-08-05 14:15:06 +08:00
d1_features.py Add files via upload 2024-08-05 14:15:06 +08:00
d2_datasets.py Add files via upload 2024-08-05 14:15:06 +08:00
m0_position.py Add files via upload 2024-08-05 14:15:06 +08:00
m1_transformer.py Add files via upload 2024-08-05 14:15:06 +08:00
m3_train.py Add files via upload 2024-08-05 14:15:06 +08:00
m4_training.py Add files via upload 2024-08-05 14:15:06 +08:00
m5_predict.py Add files via upload 2024-08-05 14:15:06 +08:00
README.md Update README.md 2024-08-05 14:16:30 +08:00
readme.txt Add files via upload 2024-08-05 14:15:06 +08:00
requirements.txt Add files via upload 2024-08-05 14:15:06 +08:00
saved_weights.pt Add files via upload 2024-08-05 14:15:06 +08:00

README.md

stock-transformer

股票人工智能模型，使用transformer模型，训练股票数据，预测未来一天的收盘价格。

步骤1： 下载数据 步骤2： 针对数据做特征处理 步骤3： 构建 dataloader，为训练做准备 步骤4： 构建模型，这里使用transformer 步骤5： 训练 步骤6： 测试或者评估（predict，evaluate），这个在测试集合上。

操作：

python m4_training.py