train -> fit update in the new KAN version

KindXiaoming · Jan 12, 2025 · 406d890 · 406d890
1 parent f871c26
commit 406d890
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);"
    ]
   },
   {