From a9e7cd5d3b6afd1f51464cb92d75c71ab2d0f62e Mon Sep 17 00:00:00 2001 From: czzhangheng Date: Thu, 20 Nov 2025 10:51:32 +0800 Subject: [PATCH] update data_selector --- dataloader/data_selector.py | 30 ++++++------------------------ 1 file changed, 6 insertions(+), 24 deletions(-) diff --git a/dataloader/data_selector.py b/dataloader/data_selector.py index 78c3e3f..3bcf3e5 100644 --- a/dataloader/data_selector.py +++ b/dataloader/data_selector.py @@ -20,40 +20,22 @@ def load_st_dataset(config): data = np.load(data_path)["data"][:, :, 0] case "PEMSD4": data_path = os.path.join("./data/PEMS04/PEMS04.npz") - data = np.load(data_path)["data"][ - :, :, 0 - ] + data = np.load(data_path)["data"][:, :, 0] case "PEMSD7": data_path = os.path.join("./data/PEMS07/PEMS07.npz") - data = np.load(data_path)["data"][ - :, :, 0 - ] + data = np.load(data_path)["data"][:, :, 0] case "PEMSD8": data_path = os.path.join("./data/PEMS08/PEMS08.npz") - data = np.load(data_path)["data"][ - :, :, 0 - ] + data = np.load(data_path)["data"][:, :, 0] case "PEMSD7(L)": data_path = os.path.join("./data/PEMS07(L)/PEMS07L.npz") - data = np.load(data_path)["data"][ - :, :, 0 - ] + data = np.load(data_path)["data"][:, :, 0] case "PEMSD7(M)": data_path = os.path.join("./data/PEMS07(M)/V_228.csv") - data = np.genfromtxt( - data_path, delimiter="," - ) - case "METR-LA": - data_path = os.path.join("./data/METR-LA/METR.h5") - with h5py.File( - data_path, "r" - ) as f: - data = np.array(f["data"]) + data = np.genfromtxt(data_path, delimiter=",") case "BJ": data_path = os.path.join("./data/BJ/BJ500.csv") - data = np.genfromtxt( - data_path, delimiter=",", skip_header=1 - ) + data = np.genfromtxt(data_path, delimiter=",", skip_header=1) case "Hainan": data_path = os.path.join("./data/Hainan/Hainan.npz") data = np.load(data_path)["data"][:, :, 0]