Save computed mapping in example

alexisthual · Mar 12, 2023 · 4c3c95f · 4c3c95f
1 parent 3111413
commit 4c3c95f
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -8,4 +8,6 @@ doc/auto_examples
 doc/gen_modules
 doc/generated
 doc/modules/generated
-.vscode
+.vscode
+
+examples/**/*.pkl
diff --git a/examples/01_brain_alignment/plot_1_aligning_brain_dense.py b/examples/01_brain_alignment/plot_1_aligning_brain_dense.py
@@ -7,8 +7,9 @@
 In this example, we align 2 low-resolution left hemispheres
 using 4 fMRI feature maps (z-score contrast maps).
 """
-
 # sphinx_gallery_thumbnail_number = 6
+import pickle
+
 import gdist
 import matplotlib as mpl
 import matplotlib.gridspec as gridspec
@@ -299,6 +300,17 @@ def compute_geometry_from_mesh(mesh_path):
     verbose=True,
 )
 
+# %%
+# Computed mappings can easily be saved on disk and loaded back:
+
+# Save mappings
+with open("./mapping.pkl", "wb") as f:
+    pickle.dump(mapping, f)
+
+# Load mappings
+with open("./mapping.pkl", "rb") as f:
+    mapping = pickle.load(f)
+
 # %%
 # Here is the evolution of the FUGW loss during training,
 # without the entropic term. Note how, in this case,