Update vanderpol.ipynb

agosztolai · Jan 3, 2024 · 4cfeede · 4cfeede
1 parent 4039a59
commit 4cfeede
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/toy_examples/vanderpol.ipynb b/examples/toy_examples/vanderpol.ipynb
@@ -184,7 +184,7 @@
     }
    ],
    "source": [
-    "data = MARBLE.construct_dataset(pos, features=vel, stop_crit=0.03)"
+    "data = MARBLE.construct_dataset(anchor=pos, vector=vel, spacing=0.03)"
    ]
   },
   {
@@ -513,7 +513,7 @@
     "mus = np.linspace(-0.1, 0.1, n_mus)\n",
     "pos, vel = get_pos_vel(mus)\n",
     "\n",
-    "data = MARBLE.construct_dataset(pos, features=vel, k=20, stop_crit=0.03)\n",
+    "data = MARBLE.construct_dataset(anchor=pos, vector=vel, k=20, spacing=0.03)\n",
     "model = MARBLE.net(data, params=params)\n",
     "model.fit(data, outdir='model_zoom')"
    ]
@@ -759,7 +759,7 @@
     "mus = np.linspace(-0.1, 0.1, n_mus)\n",
     "pos, vel = get_pos_vel_noise(mus, alpha=0.2)\n",
     "\n",
-    "data = MARBLE.construct_dataset(pos, features=vel, k=20, stop_crit=0.03)\n",
+    "data = MARBLE.construct_dataset(anchor=pos, vector=vel, k=20, spacing=0.03)\n",
     "\n",
     "params['inner_product_features'] = True\n",
     "model = MARBLE.net(data, params=params)\n",