2024-08-05 14:13:17 +08:00
|
|
|
|
# stock-transformer
|
|
|
|
|
|
股票人工智能模型，使用transformer模型，训练股票数据，预测未来一天的收盘价格。
|
2024-08-05 14:16:30 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
步骤1： 下载数据
|
|
|
|
|
|
步骤2： 针对数据做特征处理
|
|
|
|
|
|
步骤3： 构建 dataloader，为训练做准备
|
|
|
|
|
|
步骤4： 构建模型，这里使用transformer
|
|
|
|
|
|
步骤5： 训练
|
|
|
|
|
|
步骤6： 测试或者评估（predict，evaluate），这个在测试集合上。
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
操作：
|
|
|
|
|
|
|
|
|
|
|
|
python m4_training.py
|
|
|
|
|
|
|
