mirror of
https://github.com/dupenf/stock-transformer.git
synced 2024-11-25 16:08:34 +08:00
437 B
437 B
stock-transformer
股票人工智能模型,使用transformer模型,训练股票数据,预测未来一天的收盘价格。
步骤1: 下载数据 步骤2: 针对数据做特征处理 步骤3: 构建 dataloader,为训练做准备 步骤4: 构建模型,这里使用transformer 步骤5: 训练 步骤6: 测试或者评估(predict,evaluate),这个在测试集合上。
操作:
python m4_training.py