From 05da0927620309f4a0bae649202b553f6227e692 Mon Sep 17 00:00:00 2001
From: Michael Psenka <michael.psenka@gmail.com>
Date: Wed, 16 Feb 2022 23:55:44 -0800
Subject: [PATCH] now shows Z_test, demonstrates generalization

---
 examples/gaussian3d.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/gaussian3d.ipynb b/examples/gaussian3d.ipynb
index fdbd070..fe025b5 100644
--- a/examples/gaussian3d.ipynb
+++ b/examples/gaussian3d.ipynb
@@ -107,7 +107,7 @@
    "source": [
     "Z_test = net(X_test).detach()\n",
     "ue.plot_3d(X_test, y_test, 'X_test')\n",
-    "ue.plot_3d(X_test, y_test, 'Z_test')"
+    "ue.plot_3d(Z_test, y_test, 'Z_test')"
    ]
   },
   {