KindXiaoming / pykan

Kolmogorov Arnold Networks
MIT License
14.74k stars 1.35k forks source link

KAN for multi outputs? #299

Closed jacktang closed 2 months ago

jacktang commented 3 months ago

Hello, I read all examples and find that the output is single variable(univariate regression), can KAN solve multivariate regression?

jacktang commented 3 months ago

well, I find the example in https://github.com/KindXiaoming/pykan/blob/master/tutorials/physics_informed_kan.ipynb

KindXiaoming commented 2 months ago

KAN can take any width which is a list of positive integers, and the last dimension is the output dimension. E.g, KAN(width=[2,10,10,3]) has 3 output dimensions.