-
Notifications
You must be signed in to change notification settings - Fork 9
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
KAN implementation in the TEDEouS #41
Conversation
KAN implementation was added to the models.py file. The implementation consists of two classes: 1. class KANLinear - realization of a linear layer using Kolmogorov-Arnold splines. It allows to model nonlinear dependencies between input and output data using splines. 2. class KAN - realization of multilayer network using Kolmogorov-Arnold splines.
Adding an example of solving the wave equation using KAN.
Adding an example of solving the Bürgers equation using KAN.
Update model.py
atm we need to find a way out how to properly import efficient-kan to not to take all required code manually and be able to change KAN backbone if it will be the case |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It seems that we need a fork of efficient-kan with wheels ot something like this. I think, we could make PR to a main efficient-kan repo in the future.
Adding git modules: kan, efficient_kan, fast_kan
Update kan example
Create efficient kan example
Update burgers inverse kan example
Create burgers inverse example
This reverts commit 8d7a28f.
Create burgers inverse efficient kan example
Update requirements
Update requirements
Adding condition for fast kan
Adding some attributes for different models
Adding some attributes for different models
Update efficient kan example
Update kan parameters in wave physics kan example
Create wave physics fast kan example
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
We need to fix last one and we're good to go on this PR.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It seems that we do not need code here anymore.
Delete custom KAN
Update parameters
Update parameters
Update parameters
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Okay, we're now good to go.
1. Change models.py file:
KAN implementation was added to the models.py file. The implementation consists of two classes:
2. Adding examples of equation solutions: