From 730a3a63ecf8e30590823e3376bd3d9f4bbd4834 Mon Sep 17 00:00:00 2001 From: tuanzi Date: Sun, 21 Jul 2024 12:19:33 +0800 Subject: [PATCH] [Fix] Update API_2_plotting.ipynb to make it runs correctly --- tutorials/API_2_plotting.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tutorials/API_2_plotting.ipynb b/tutorials/API_2_plotting.ipynb index 19ed3e99..acc57755 100644 --- a/tutorials/API_2_plotting.ipynb +++ b/tutorials/API_2_plotting.ipynb @@ -36,7 +36,7 @@ "source": [ "from kan import *\n", "# create a KAN: 2D inputs, 1D output, and 5 hidden neurons. cubic spline (k=3), 5 grid intervals (grid=5).\n", - "model = KAN(width=[2,5,1], grid=3, k=3, seed=1)\n", + "model = KAN(width=[2,5,1], grid=3, k=3, seed=1, auto_save=True)\n", "\n", "# create dataset f(x,y) = exp(sin(pi*x)+y^2)\n", "f = lambda x: torch.exp(torch.sin(torch.pi*x[:,[0]]) + x[:,[1]]**2)\n", @@ -121,7 +121,7 @@ ], "source": [ "# train the model\n", - "model.fit(dataset, opt=\"LBFGS\", steps=20, lamb=0.01, auto_save=True);" + "model.fit(dataset, opt=\"LBFGS\", steps=20, lamb=0.01);" ] }, {