From 406d8904f2be3e49246c7279655680943799d200 Mon Sep 17 00:00:00 2001 From: Dmitrii Dulaev <37038612+Timoniche@users.noreply.github.com> Date: Mon, 13 Jan 2025 00:23:39 +0300 Subject: [PATCH] train -> fit update in the new KAN version --- docs/intro.ipynb | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/intro.ipynb b/docs/intro.ipynb index 0d7e1de..c9a10dd 100644 --- 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);" ] }, {