.1189032313505632:7037e6b954a99ae4425c8d3d88e92158_69b6117428ff963a2f47d9ae.69b6164a28ff963a2f47da68.69b616434b2b6785f1fbb84b:Trae CN.T(2026/3/15 10:15:38)

download_dataset.py

@@ -0,0 +1,14 @@
+from datasets import load_dataset
+import pandas as pd
+import os
+
+# 确保dataset目录存在
+os.makedirs('dataset', exist_ok=True)
+
+# 加载ETTh1数据集
+dataset = load_dataset('thuml/Time-Series-Library', 'ETTh1')
+
+# 将训练集保存为CSV
+print("Saving ETTh1 dataset...")
+dataset['train'].to_csv('dataset/ETTh1.csv', index=False)
+print("Dataset saved successfully!")

extract_zips.py

@@ -0,0 +1,16 @@
+import zipfile
+import os
+from pathlib import Path
+
+def extract_all_zips(root_dir):
+    for path in Path(root_dir).rglob('*.zip'):
+        print(f"Extracting {path}...")
+        try:
+            with zipfile.ZipFile(path, 'r') as zip_ref:
+                zip_ref.extractall(path.parent)
+            print(f"Successfully extracted {path}")
+        except Exception as e:
+            print(f"Error extracting {path}: {e}")
+
+if __name__ == "__main__":
+    extract_all_zips('dataset')