folivo/thermal-conduction-ml
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add learnable finite difference and fix __init__.py

Browse Source
This commit is contained in:
Filippo Olivo
2025-10-27 10:54:35 +01:00
parent 6e90ef5393
commit 35770ace67
3 changed files with 162 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
ThermalSolver/model/__init__.py
View File

@@ -1,5 +1,13 @@
__all__ = ["GraphFiniteDifference", "GatingGNO"]
__all__ = [
"GraphFiniteDifference",
"GatingGNO",
"LearnableGraphFiniteDifference",
"PointNet",
]
from .learnable_finite_difference import GraphFiniteDifference
from .learnable_finite_difference import (
GraphFiniteDifference as LearnableGraphFiniteDifference,
)
from .finite_difference import GraphFiniteDifference as GraphFiniteDifference
from .local_gno import GatingGNO
from .point_net import PointNet