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 FilippoOlivo
parent ff191c9611
commit 01aeb17673
4 changed files with 147 additions and 0 deletions

View File

@@ -0,0 +1,9 @@
NeuralTangentKernelWeighting
=============================
.. currentmodule:: pina.loss.ntk_weighting
.. automodule:: pina.loss.ntk_weighting
.. autoclass:: NeuralTangentKernelWeighting
:members:
:show-inheritance: