Upload 3 files

Exp3_Kuroshio_forecasting/S3_results/forecast_merged.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:90f67807503557deed914aa2c3d046fb12071a2a1edac5289ab564af39a96415
+size 1378960303

Exp3_Kuroshio_forecasting/S3_results/merge.ipynb

@@ -0,0 +1,157 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "id": "3f7c354c-4f10-4020-a303-3b8a8e612468",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Merging forecast_20210101.h5\n",
+      "Merging forecast_20210106.h5\n",
+      "Merging forecast_20210111.h5\n",
+      "Merging forecast_20210116.h5\n",
+      "Merging forecast_20210121.h5\n",
+      "Merging forecast_20210126.h5\n",
+      "Merging forecast_20210131.h5\n",
+      "Merging forecast_20210205.h5\n",
+      "Merging forecast_20210210.h5\n",
+      "Merging forecast_20210215.h5\n",
+      "Merging forecast_20210220.h5\n",
+      "Merging forecast_20210225.h5\n",
+      "Merging forecast_20210302.h5\n",
+      "Merging forecast_20210307.h5\n",
+      "Merging forecast_20210312.h5\n",
+      "Merging forecast_20210317.h5\n",
+      "Merging forecast_20210322.h5\n",
+      "Merging forecast_20210327.h5\n",
+      "Merging forecast_20210401.h5\n",
+      "Merging forecast_20210406.h5\n",
+      "Merging forecast_20210411.h5\n",
+      "Merging forecast_20210416.h5\n",
+      "Merging forecast_20210421.h5\n",
+      "Merging forecast_20210426.h5\n",
+      "Merging forecast_20210501.h5\n",
+      "Merging forecast_20210506.h5\n",
+      "Merging forecast_20210511.h5\n",
+      "Merging forecast_20210516.h5\n",
+      "Merging forecast_20210521.h5\n",
+      "Merging forecast_20210526.h5\n",
+      "Merging forecast_20210531.h5\n",
+      "Merging forecast_20210605.h5\n",
+      "Merging forecast_20210610.h5\n",
+      "Merging forecast_20210615.h5\n",
+      "Merging forecast_20210620.h5\n",
+      "Merging forecast_20210625.h5\n",
+      "Merging forecast_20210630.h5\n",
+      "Merging forecast_20210705.h5\n",
+      "Merging forecast_20210710.h5\n",
+      "Merging forecast_20210715.h5\n",
+      "Merging forecast_20210720.h5\n",
+      "Merging forecast_20210725.h5\n",
+      "Merging forecast_20210730.h5\n",
+      "Merging forecast_20210804.h5\n",
+      "Merging forecast_20210809.h5\n",
+      "Merging forecast_20210814.h5\n",
+      "Merging forecast_20210819.h5\n",
+      "Merging forecast_20210824.h5\n",
+      "Merging forecast_20210829.h5\n",
+      "Merging forecast_20210903.h5\n",
+      "Merging forecast_20210908.h5\n",
+      "Merging forecast_20210913.h5\n",
+      "Merging forecast_20210918.h5\n",
+      "Merging forecast_20210923.h5\n",
+      "Merging forecast_20210928.h5\n",
+      "Merging forecast_20211003.h5\n",
+      "Merging forecast_20211008.h5\n",
+      "Merging forecast_20211013.h5\n",
+      "Merging forecast_20211018.h5\n",
+      "Merging forecast_20211023.h5\n",
+      "Merging forecast_20211028.h5\n",
+      "Merging forecast_20211102.h5\n",
+      "Merging forecast_20211107.h5\n",
+      "Merging forecast_20211112.h5\n",
+      "Merging forecast_20211117.h5\n",
+      "Merging forecast_20211122.h5\n",
+      "Merging forecast_20211127.h5\n",
+      "Merging forecast_20211202.h5\n",
+      "Merging forecast_20211207.h5\n",
+      "Merging forecast_20211212.h5\n",
+      "Merging forecast_20211217.h5\n",
+      "Merging forecast_20211222.h5\n",
+      "Merging forecast_20211227.h5\n",
+      "合并完成！\n",
+      "merged_label shape: (73, 120, 2, 128, 128)\n",
+      "merged_prediction shape: (73, 120, 2, 128, 128)\n"
+     ]
+    }
+   ],
+   "source": [
+    "import os\n",
+    "import h5py\n",
+    "import numpy as np\n",
+    "\n",
+    "def merge_label_prediction(output_file='forecast_merged.h5'):\n",
+    "    # 1) 找出当前目录下所有 forecast_*.h5 文件，并按文件名排序\n",
+    "    file_list = sorted([\n",
+    "        f for f in os.listdir('.')\n",
+    "        if f.startswith('forecast_') and f.endswith('.h5')\n",
+    "    ])\n",
+    "\n",
+    "    if len(file_list) == 0:\n",
+    "        print(\"未找到任何 forecast_*.h5 文件！\")\n",
+    "        return\n",
+    "\n",
+    "    all_label = []\n",
+    "    all_prediction = []\n",
+    "\n",
+    "    # 2) 逐个读取并收集 label、prediction\n",
+    "    for fname in file_list:\n",
+    "        print(f\"Merging {fname}\")\n",
+    "        with h5py.File(fname, 'r') as f:\n",
+    "            all_label.append(f['label'][:])         # shape 比如 (1, 120, 2, 128, 128)\n",
+    "            all_prediction.append(f['prediction'][:])\n",
+    "\n",
+    "    # 3) 沿着第0维拼接 (把原先每个文件中的前置“1”给堆叠起来)\n",
+    "    merged_label = np.concatenate(all_label, axis=0)\n",
+    "    merged_prediction = np.concatenate(all_prediction, axis=0)\n",
+    "\n",
+    "    # 4) 写出合并结果到新文件\n",
+    "    with h5py.File(output_file, 'w') as f_out:\n",
+    "        f_out.create_dataset('label', data=merged_label, compression='gzip')\n",
+    "        f_out.create_dataset('prediction', data=merged_prediction, compression='gzip')\n",
+    "\n",
+    "    print(\"合并完成！\")\n",
+    "    print(\"merged_label shape:\", merged_label.shape)\n",
+    "    print(\"merged_prediction shape:\", merged_prediction.shape)\n",
+    "\n",
+    "if __name__ == \"__main__\":\n",
+    "    merge_label_prediction()\n"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.15"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}