改良KAN模型的PyTorch完整源码，Kolmogorov-Arnold Networks
5星
- 浏览量: 0
- 大小:None
- 文件类型：ZIP
简介：
本项目提供了一个基于PyTorch框架的改进版KAN（Kolmogorov-Arnold Networks）模型的完整源代码，适用于深度学习研究与实践。
改进KAN模型的Pytorch完整源码如下所示：
```python
class MNISTChebyKAN(nn.Module):
def __init__(self):
super(MNISTChebyKAN, self).__init__()
self.chebykan1 = ChebyKANLayer(28*28, 32, 4)
self.ln1 = nn.LayerNorm(32) # 使用 LayerNorm 来避免 tanh 函数造成的梯度消失问题
self.chebykan2 = ChebyKANLayer(32, 16, 4)
self.ln2 = nn.LayerNorm(16)
self.chebykan3 = ChebyKANLayer(16, 10, 4)
```
这里，`Chebyshev polynomials` 替代了 `B-splines`。
全部评论 (0)
还没有任何评论哟~
