时间序列预测 | Python实现Attention-Transformer时间序列预测

• main.py

import argparse
from collections import defaultdict
from dataset_TSAT_ETTm1_48 import generate_ETTm1_data
from dataset_graph import construct_TSAT_dataset, graph_collate_func_TSAT_normalization_require
import numpy as np
from sklearn.model_selection import