Skip to content

Commit

Permalink
[docs] Add auto API generator
Browse files Browse the repository at this point in the history
  • Loading branch information
Routhleck committed Dec 13, 2024
1 parent 28f1d10 commit ebce306
Show file tree
Hide file tree
Showing 17 changed files with 1,070 additions and 27 deletions.
3 changes: 2 additions & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -30,4 +30,5 @@ related-projects.md
docs/**/*.dat
docs/**/*.npz

.DS_Store
.DS_Store
docs/apis/generated/
9 changes: 9 additions & 0 deletions docs/_templates/classtemplate.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
.. role:: hidden
:class: hidden-section
.. currentmodule:: {{ module }}


{{ name | underline}}

.. autoclass:: {{ name }}
:members:
26 changes: 26 additions & 0 deletions docs/apis/pinnx.callbacks.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
``pinnx.callbacks`` module
==========================

.. currentmodule:: pinnx.callbacks
.. automodule:: pinnx.callbacks

Callbacks
---------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

Callback
CallbackList
ModelCheckpoint
EarlyStopping
Timer
DropoutUncertainty
VariableValue
OperatorPredictor
MovieDumper
PDEPointResampler


23 changes: 23 additions & 0 deletions docs/apis/pinnx.fnspace.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
``pinnx.fnspace`` module
========================

.. currentmodule:: pinnx.fnspace
.. automodule:: pinnx.fnspace

Function Space
--------------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

wasserstein2
FunctionSpace
PowerSeries
Chebyshev
GRF
GRF_KL
GRF2D


117 changes: 117 additions & 0 deletions docs/apis/pinnx.geometry.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,117 @@
``pinnx.geometry`` module
=========================

.. currentmodule:: pinnx.geometry
.. automodule:: pinnx.geometry

Base Geometry Class
-------------------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

AbstractGeometry
Geometry
CSGUnion
CSGDifference
CSGIntersection


Geometry in 1D
--------------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

sample
Geometry
Interval
Literal
Union


Geometry in 2D
--------------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

Disk
Ellipse
Polygon
Rectangle
StarShaped
Triangle


Geometry in 3D
--------------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

Rectangle
Hypercube
Hypersphere
Cuboid
Sphere
Union
Literal


Geometry in ND
--------------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

sample
isclose
Geometry
Hypercube
Hypersphere
Literal


Point Cloud
-----------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

isclose
BatchSampler
Geometry
PointCloud


Time Domain
-----------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

isclose
AbstractGeometry
Interval
Rectangle
Cuboid
Hypercube
TimeDomain
GeometryXTime


19 changes: 19 additions & 0 deletions docs/apis/pinnx.grad.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
``pinnx.grad`` module
=====================

.. currentmodule:: pinnx.grad
.. automodule:: pinnx.grad

Automatic Differentiation
-------------------------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

jacobian
hessian
gradient


50 changes: 50 additions & 0 deletions docs/apis/pinnx.icbc.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,50 @@
``pinnx.icbc`` module
=====================

.. currentmodule:: pinnx.icbc
.. automodule:: pinnx.icbc

Base Initial and Boundary Conditions Class
------------------------------------------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

AbstractGeometry
ICBC
Optional
Dict


Boundary Conditions
-------------------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

BC
DirichletBC
Interface2DBC
NeumannBC
OperatorBC
PeriodicBC
PointSetBC
PointSetOperatorBC
RobinBC


Initial Conditions
------------------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

IC


24 changes: 24 additions & 0 deletions docs/apis/pinnx.metrics.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
``pinnx.metrics`` module
========================

.. currentmodule:: pinnx.metrics
.. automodule:: pinnx.metrics

Metrics
-------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

accuracy
l2_relative_error
nanl2_relative_error
mean_l2_relative_error
mean_squared_error
mean_absolute_percentage_error
max_absolute_percentage_error
absolute_percentage_error_std


108 changes: 108 additions & 0 deletions docs/apis/pinnx.nn.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,108 @@
``pinnx.nn`` module
===================

.. currentmodule:: pinnx.nn
.. automodule:: pinnx.nn

Base Neural Network Class
-------------------------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

NN
Optional
Callable


Dict and Array Converters
-------------------------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

DictToArray
ArrayToDict


DeepONet Strategy
-----------------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

abstractmethod
ABC
DeepONetStrategy
SingleOutputStrategy
IndependentStrategy
SplitBothStrategy
SplitBranchStrategy
SplitTrunkStrategy


DeepONet
--------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

DeepONet
DeepONetCartesianProd
PODDeepONet


Fully Connected Neural Network
------------------------------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

get_activation
NN
FNN
PFNN
Union
Callable
Sequence
Optional


Multiple Input Operators Network
--------------------------------

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

get_activation
NN
FNN
MIONetCartesianProd
PODMIONet
Optional
Callable


Model
-----

.. autosummary::
:toctree: generated/
:nosignatures:
:template: classtemplate.rst

Model


Loading

0 comments on commit ebce306

Please sign in to comment.