update

xuhuihuang · Jan 22, 2024 · c92c5f3 · c92c5f3
1 parent 0979a0d
commit c92c5f3
Show file tree

Hide file tree

Showing 6 changed files with 526 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -48,7 +48,7 @@ python ./ts-dart/scripts/train_tsdart.py \
     --n_epochs 20 \
     --train_split 0.9 \
     --train_batch_size 1000 \
-    --data_directory ./ts-dart/data \
+    --data_directory ./ts-dart/data/quadruple-well \
     --saving_directory . 
 ```
 

diff --git a/data/muller/muller.npy b/data/muller/muller.npy
diff --git a/data/quadruple-well.npy → data/quadruple-well/quadruple-well.npy b/data/quadruple-well.npy → data/quadruple-well/quadruple-well.npy
diff --git a/example/muller-example.ipynb b/example/muller-example.ipynb
diff --git a/example/quadruple-well-example.ipynb b/example/quadruple-well-example.ipynb
@@ -126,7 +126,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "data = np.load('../data/quadruple-well.npy')\n",
+    "data = np.load('../data/quadruple-well/quadruple-well.npy')\n",
     "\n",
     "pre = Preprocessing(dtype=np.float32)\n",
     "dataset = pre.create_dataset(lag_time=10,data=data)"

diff --git a/scripts/train_tsdart.sh b/scripts/train_tsdart.sh
@@ -14,5 +14,5 @@ python ./ts-dart/scripts/train_tsdart.py \
     --n_epochs 20 \
     --train_split 0.9 \
     --train_batch_size 1000 \
-    --data_directory ./ts-dart/data \
+    --data_directory ./ts-dart/data/quadruple-well \
     --saving_directory .