updating clustering example in tutorial

htjb · Aug 4, 2023 · 90c74f7 · 90c74f7
1 parent 6c7ab87
commit 90c74f7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/notebook/Tutorial.ipynb b/notebook/Tutorial.ipynb
@@ -228,7 +228,7 @@
    "source": [
     "# Clustering with margarine\n",
     "\n",
-    "The below example demonstrates how to perform clustering with `margarine` which can improve the accuracy of the emulation."
+    "The below example demonstrates how to perform clustering with `margarine` which can improve the accuracy of the emulation. The clustering class has all the same functionality as the MAF class."
    ]
   },
   {
@@ -254,7 +254,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "flow = MAF(data, np.ones(len(data)), clustering=True)\n",
+    "from margarine.clustered import clusterMAF\n",
+    "flow = clusterMAF(data, np.ones(len(data)))\n",
     "flow.train(10000, early_stop=True)"
    ]
   },