search

KindXiaoming / pykan

Kolmogorov Arnold Networks
MIT License
13.84k stars 1.24k forks source link

model(dataset['train_input']) #294

Closed Papillon-forest closed 5 days ago

Papillon-forest commented 2 weeks ago

我想知道下列两段代码的区别在哪，或许有人可以点拨我一下吗？

model = model.prune()
model(dataset['train_input'])
model.plot()
model = model.prune()
model.plot()
KindXiaoming commented 2 weeks ago

上面那个可以跑通，下面那个会报错。主要是Plot之前需要forward一次获取activation。

Papillon-forest commented 2 weeks ago

@KindXiaoming Thank you for your guidance in your busy schedule, it is much appreciated!