fix: 统一 NPZ 命名格式,添加 NetCDF ZIP 解压工具和 h5netcdf/h5py 依赖
This commit is contained in:
@@ -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"])
|
||||
|
||||
Reference in New Issue
Block a user