【kan】hello, kan
创始人
2024-11-12 20:05:25
0

目录

安装

初始化kan

创建数据集

训练KAN

剪枝 KAN

继续训练

设置公式

继续训练

获得符号公式


安装

pip install pykan

初始化kan

from kan import * torch.set_default_dtype(torch.float64) # 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=3, k=3, seed=42)
checkpoint directory created: ./model saving model version 0.0

创建数据集

from kan.utils import create_dataset # create dataset f(x,y) = exp(sin(pi*x)+y^2) f = lambda x: torch.exp(torch.sin(torch.pi*x[:,[0]]) + x[:,[1]]**2) dataset = create_dataset(f, n_var=2) dataset['train_input'].shape, dataset['train_label'].shape
(torch.Size([1000, 2]), torch.Size([1000, 1]))

相关内容

热门资讯

wepoke辅助德之星!wep... wepoke辅助德之星!wepoker都是真的有挂(透视)果真真的有挂(2020已更新)(哔哩哔哩)...
德州ai!wepoke有规律(... 德州ai!wepoke有规律(透明黑科技)苹果版教程-果然真的有挂(百度贴吧)1、下载好德州ai辅助...
七分钟揭秘!impoker德州... 七分钟揭秘!impoker德州外挂,(德州ai)一般真的有挂,辅助教程(有挂精通)1、下载好impo...
一分钟秒懂!wopoker有外... 一分钟秒懂!wopoker有外挂(透明黑科技)wepowerai辅助工具-果真真的有挂(知乎)所有人...
微扑克wpk透视辅助!impo... 微扑克wpk透视辅助!impoker都是真的有挂(透视)一直真的有挂(2023已更新)(哔哩哔哩);...
菠萝德州!wepoke挂透视(... 菠萝德州!wepoke挂透视(黑科技)插件挂教程-果然真的有挂(头条);1、进入游戏-大厅左侧-新手...
玩家必看科普!微扑克辅助安卓版... 玩家必看科普!微扑克辅助安卓版本(软件透明挂)wepOKE挂被实锤-果真真的有挂(微博热搜)进入游戏...
五分钟科研!aapoker德州... 五分钟科研!aapoker德州俱乐部,(德州ai)好像真的有挂,2025新版技巧(有挂晓得)aapo...
wpk有辅助挂!we-poke... wpk有辅助挂!we-poker原来真的有挂(软件透明挂)好像真的有挂(2025已更新)(哔哩哔哩)...
we辅助poker!wepok... we辅助poker!wepoke苹果版挂(辅助挂)最新版教程-果真真的有挂(今日头条)1、金币登录送...