Merge pull request #512 from Timoniche/patch-1

train -> fit update in the new KAN version
KindXiaoming · Jan 12, 2025 · 0a452a0 · 0a452a0
2 parents f871c26 + 406d890
commit 0a452a0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/docs/intro.ipynb b/docs/intro.ipynb
@@ -208,7 +208,7 @@
    ],
    "source": [
     "# train the model\n",
-    "model.train(dataset, opt=\"LBFGS\", steps=20, lamb=0.01, lamb_entropy=10.);"
+    "model.fit(dataset, opt=\"LBFGS\", steps=20, lamb=0.01, lamb_entropy=10.);"
    ]
   },
   {
@@ -324,7 +324,7 @@
     }
    ],
    "source": [
-    "model.train(dataset, opt=\"LBFGS\", steps=50);"
+    "model.fit(dataset, opt=\"LBFGS\", steps=50);"
    ]
   },
   {
@@ -409,7 +409,7 @@
     }
    ],
    "source": [
-    "model.train(dataset, opt=\"LBFGS\", steps=50);"
+    "model.fit(dataset, opt=\"LBFGS\", steps=50);"
    ]
   },
   {