Files
PINA/pina/model/__init__.py
ajacoby9 0844282335 Improve and fix spline (#610)
* vectorize Cox - de Boor recursion

Co-authored-by: Filippo Olivo <folivo@filippoolivo.com>
Co-authored-by: ajacoby9 <a99jacoby@gmail.com>

* fix logic and extend tests

* add b-spline surface

---------

Co-authored-by: Filippo Olivo <folivo@filippoolivo.com>
Co-authored-by: GiovanniCanali <giovanni.canali98@yahoo.it>
2025-10-15 12:31:43 +02:00

34 lines
1.0 KiB
Python

"""Module for the Neural model classes."""
__all__ = [
"FeedForward",
"ResidualFeedForward",
"MultiFeedForward",
"DeepONet",
"MIONet",
"FNO",
"FourierIntegralKernel",
"KernelNeuralOperator",
"AveragingNeuralOperator",
"LowRankNeuralOperator",
"Spline",
"GraphNeuralOperator",
"PirateNet",
"EquivariantGraphNeuralOperator",
"SINDy",
]
from .feed_forward import FeedForward, ResidualFeedForward
from .multi_feed_forward import MultiFeedForward
from .deeponet import DeepONet, MIONet
from .fourier_neural_operator import FNO, FourierIntegralKernel
from .kernel_neural_operator import KernelNeuralOperator
from .average_neural_operator import AveragingNeuralOperator
from .low_rank_neural_operator import LowRankNeuralOperator
from .spline import Spline
from .spline_surface import SplineSurface
from .graph_neural_operator import GraphNeuralOperator
from .pirate_network import PirateNet
from .equivariant_graph_neural_operator import EquivariantGraphNeuralOperator
from .sindy import SINDy