Neural Tangent Kernel integration + typo fix (#505)

* NTK weighting + typo fixing
* black code formatter + .rst docs

---------

Co-authored-by: Dario Coscia <dariocos99@gmail.com>
This commit is contained in:
Giuseppe Alessio D'Inverno
2025-03-19 12:44:07 -04:00
committed by Nicola Demo
parent 2b09cb95cf
commit 716d43f146
4 changed files with 147 additions and 0 deletions

View File

@@ -6,6 +6,7 @@ __all__ = [
"PowerLoss",
"WeightingInterface",
"ScalarWeighting",
"NeuralTangentKernelWeighting",
]
from .loss_interface import LossInterface
@@ -13,3 +14,4 @@ from .power_loss import PowerLoss
from .lp_loss import LpLoss
from .weighting_interface import WeightingInterface
from .scalar_weighting import ScalarWeighting
from .ntk_weighting import NeuralTangentKernelWeighting