KAN

KAN copied to clipboard

Error running code

2

Hello. Thank you very much for sharing your KAN neural networks example. Running the following cell of your regression example notebook results = model.train(calhous_dataset, opt="LBFGS", device=device, metrics=(train_mse, test_mse), loss_fn=torch.nn.MSELoss(), steps=25,...

diego-uva

When I ran the follow commands: image_folder = 'video_img' model = KAN(width=[4, 5, 3], grid=5, k=3, seed=0, device=device) model(iris_dataset['train_input']) model.plot(beta=100, scale=1, in_vars=['SL', 'SW', 'PL', 'PW'], out_vars=['Set', 'Ver', 'Vir']) it says...

ChenHuang94