Merge pull request #512 from Timoniche/patch-1

train -> fit update in the new KAN version
This commit is contained in:
Ziming Liu 2025-01-12 17:00:03 -05:00 committed by GitHub
commit 0a452a0739
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);"
]
},
{