716d43f146
* NTK weighting + typo fixing * black code formatter + .rst docs --------- Co-authored-by: Dario Coscia <dariocos99@gmail.com>
18 lines
468 B
Python
18 lines
468 B
Python
"""Module for loss functions and weighting functions."""
|
|
|
|
__all__ = [
|
|
"LossInterface",
|
|
"LpLoss",
|
|
"PowerLoss",
|
|
"WeightingInterface",
|
|
"ScalarWeighting",
|
|
"NeuralTangentKernelWeighting",
|
|
]
|
|
|
|
from .loss_interface import LossInterface
|
|
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
|