Merge pull request #512 from Timoniche/patch-1

KindXiaoming · web-flow · commit 0a452a0739b4 · 2025-01-12T17:00:03.000-05:00
train -&gt; fit update in the new KAN version
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);"
    ]
   },
   {

Original file line number	Diff line number	Diff line change
`@@ -208,7 +208,7 @@`
`208`	`208`	`],`
`209`	`209`	`"source": [`
`210`	`210`	`"# train the model\n",`
`211`		`- "model.train(dataset, opt=\"LBFGS\", steps=20, lamb=0.01, lamb_entropy=10.);"`
	`211`	`+ "model.fit(dataset, opt=\"LBFGS\", steps=20, lamb=0.01, lamb_entropy=10.);"`
`212`	`212`	`]`
`213`	`213`	`},`
`214`	`214`	`{`
`@@ -324,7 +324,7 @@`
`324`	`324`	`}`
`325`	`325`	`],`
`326`	`326`	`"source": [`
`327`		`- "model.train(dataset, opt=\"LBFGS\", steps=50);"`
	`327`	`+ "model.fit(dataset, opt=\"LBFGS\", steps=50);"`
`328`	`328`	`]`
`329`	`329`	`},`
`330`	`330`	`{`
`@@ -409,7 +409,7 @@`
`409`	`409`	`}`
`410`	`410`	`],`
`411`	`411`	`"source": [`
`412`		`- "model.train(dataset, opt=\"LBFGS\", steps=50);"`
	`412`	`+ "model.fit(dataset, opt=\"LBFGS\", steps=50);"`
`413`	`413`	`]`
`414`	`414`	`},`
`415`	`415`	`{`