项目演示:https://www.bilibili.com/video/BV1Wg4y1u7aZ/?spm_id_from=333.999.0.0
结果展示:
# Transformer的交通流量预测 完整数据代码 整合了时间天气等多方面信息
import datetime
from torch.utils.data import DataLoader
from dataset import TrafficDataset
from model import TransAm, LSTMModel, lstm_init_weights, Transformer
import argparse
import torch
import torch.nn as nn
from tqdm import tqdm
import math
from utils import PlotUtils
from constant impo