Updated examples.

OHBA-analysis · Mar 26, 2024 · 20ffce9 · 20ffce9
1 parent 6d51e24
commit 20ffce9
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 4 deletions.
diff --git a/examples/fmri/biobank/2_train_hmm.py b/examples/fmri/biobank/2_train_hmm.py
@@ -35,7 +35,12 @@
     inputs = file.read().split("\n")
 
 # Create Data object for training
-data = Data(inputs, load_memmaps=False, n_jobs=8)
+data = Data(
+    inputs,
+    use_tfrecord=True,
+    store_dir=f"tmp_{id}",
+    n_jobs=8,
+)
 
 # Prepare data
 data.standardize()

diff --git a/examples/fmri/biobank/4_dual_estimation.py b/examples/fmri/biobank/4_dual_estimation.py
@@ -47,7 +47,12 @@
     inputs = file.read().split("\n")
 
 # Create Data object for training
-data = Data(inputs, load_memmaps=False, n_jobs=8)
+data = Data(
+    inputs,
+    use_tfrecord=True,
+    store_dir=f"tmp_{id}",
+    n_jobs=16,
+)
 
 # Prepare data
 data.standardize()

diff --git a/examples/fmri/hcp/2_train_hmm.py b/examples/fmri/hcp/2_train_hmm.py
@@ -35,7 +35,11 @@
     inputs = file.read().split("\n")
 
 # Create Data object for training
-data = Data(inputs, load_memmaps=False, n_jobs=8)
+data = Data(
+    inputs,
+    store_dir=f"tmp_{id}",
+    n_jobs=8,
+)
 
 # Prepare data
 data.standardize()

diff --git a/examples/fmri/hcp/4_dual_estimation.py b/examples/fmri/hcp/4_dual_estimation.py
@@ -47,7 +47,11 @@
     inputs = file.read().split("\n")
 
 # Create Data object for training
-data = Data(inputs, load_memmaps=False, n_jobs=8)
+data = Data(
+    inputs,
+    store_dir=f"tmp_{id}",
+    n_jobs=8,
+)
 
 # Prepare data
 data.standardize()