train -> fit update in the new KAN version

This commit is contained in:
Dmitrii Dulaev 2025-01-13 00:23:39 +03:00 committed by GitHub
parent f871c26d4d
commit 406d8904f2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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);"
]
},
{