【KAN】【API教程】get_fun

抽取某个激活函数的样子

from kan import *
import matplotlib.pyplot as plt
# create a KAN: 2D inputs, 1D output, and 5 hidden neurons. cubic spline (k=3), 5 grid intervals (grid=5).
model = KAN(width=[2,5,1], grid=5, k=3, seed=0)
x = torch.normal(0,1,size=(100,2))
model(x)
model.plot(beta=100)

l = 0
i = 0
j = 3
x, y = model.get_fun(l,i,j)