fix: 统一 NPZ 命名格式,添加 NetCDF ZIP 解压工具和 h5netcdf/h5py 依赖

This commit is contained in:
2026-05-28 08:26:18 +08:00
parent 07468266b4
commit a2c9ba4863
3 changed files with 66 additions and 2 deletions
+2 -2
View File
@@ -534,7 +534,7 @@ def preprocess_all() -> None:
if not saved_feature_cols:
saved_feature_cols = feature_cols
npz_path = DATA_PROCESSED / f"sequences_{city_key}.npz"
npz_path = DATA_PROCESSED / f"{city_key}_sequences.npz"
np.savez_compressed(
npz_path,
X=X,
@@ -561,7 +561,7 @@ def preprocess_all() -> None:
# 合并 NPZ
all_X, all_y = [], []
for city_key in CITIES:
npz_path = DATA_PROCESSED / f"sequences_{city_key}.npz"
npz_path = DATA_PROCESSED / f"{city_key}_sequences.npz"
if npz_path.exists():
data = np.load(npz_path, allow_pickle=True)
all_X.append(data["X"])