deploy: 80d05aa

_sources/autoapi/lasdi/gplasdi/index.rst.txt

@@ -116,6 +116,26 @@ Module Contents
 
 
 
+   .. py:attribute:: training_loss
+      :value: []
+
+
+
+   .. py:attribute:: ae_loss
+      :value: []
+
+
+
+   .. py:attribute:: ld_loss
+      :value: []
+
+
+
+   .. py:attribute:: coef_loss
+      :value: []
+
+
+
    .. py:method:: train()
 
 

_sources/autoapi/lasdi/index.rst.txt

@@ -17,6 +17,7 @@ Submodules
    /autoapi/lasdi/inputs/index
    /autoapi/lasdi/latent_dynamics/index
    /autoapi/lasdi/latent_space/index
+   /autoapi/lasdi/networks/index
    /autoapi/lasdi/param/index
    /autoapi/lasdi/physics/index
    /autoapi/lasdi/postprocess/index

_sources/autoapi/lasdi/latent_space/index.rst.txt

@@ -4,20 +4,13 @@ lasdi.latent_space
 .. py:module:: lasdi.latent_space
 
 
-Attributes
-----------
-
-.. autoapisummary::
-
-   lasdi.latent_space.act_dict
-
-
 Classes
 -------
 
 .. autoapisummary::
 
    lasdi.latent_space.MultiLayerPerceptron
+   lasdi.latent_space.LatentSpace
    lasdi.latent_space.Autoencoder
 
 
@@ -29,18 +22,10 @@ Functions
    lasdi.latent_space.initial_condition_latent
 
 
-Module Contents
----------------
-
-.. py:data:: act_dict
-
-.. py:function:: initial_condition_latent(param_grid, physics, autoencoder)
-
-   Outputs the initial condition in the latent space: Z0 = encoder(U0)
-
-
+Package Contents
+----------------
 
-.. py:class:: MultiLayerPerceptron(layer_sizes, act_type='sigmoid', reshape_index=None, reshape_shape=None, threshold=0.1, value=0.0, num_heads=1)
+.. py:class:: MultiLayerPerceptron(layer_sizes, act_type='sigmoid', reshape_index=None, reshape_shape=None, threshold=0.1, value=0.0)
 
    Bases: :py:obj:`torch.nn.Module`
 
@@ -65,32 +50,49 @@ Module Contents
    .. py:attribute:: act_type
 
 
-   .. py:attribute:: use_multihead
-      :value: False
-
+   .. py:method:: forward(x)
 
 
-   .. py:method:: forward(x)
+   .. py:method:: init_weight()
 
 
-   .. py:method:: apply_attention(x, act_idx)
+.. py:function:: initial_condition_latent(param_grid, physics, autoencoder)
 
+   Outputs the initial condition in the latent space: Z0 = encoder(U0)
 
-   .. py:method:: init_weight()
 
 
-.. py:class:: Autoencoder(physics, config)
+.. py:class:: LatentSpace(physics, config)
 
    Bases: :py:obj:`torch.nn.Module`
 
 
    .. py:attribute:: qgrid_size
 
 
-   .. py:attribute:: space_dim
+   .. py:attribute:: n_z
 
 
-   .. py:attribute:: n_z
+   .. py:method:: forward(x)
+
+
+   .. py:method:: export()
+
+
+   .. py:method:: load(dict_)
+
+      .. rubric:: Notes
+
+      This abstract class only checks if the variables in restart file are the same as the instance attributes.
+
+
+
+.. py:class:: Autoencoder(physics, config)
+
+   Bases: :py:obj:`LatentSpace`
+
+
+   .. py:attribute:: space_dim
 
 
    .. py:attribute:: encoder
@@ -107,4 +109,9 @@ Module Contents
 
    .. py:method:: load(dict_)
 
+      .. rubric:: Notes
+
+      This abstract class only checks if the variables in restart file are the same as the instance attributes.
+
+
 

_sources/autoapi/lasdi/networks/index.rst.txt

@@ -0,0 +1,155 @@
+lasdi.networks
+==============
+
+.. py:module:: lasdi.networks
+
+
+Attributes
+----------
+
+.. autoapisummary::
+
+   lasdi.networks.act_dict
+
+
+Classes
+-------
+
+.. autoapisummary::
+
+   lasdi.networks.MultiLayerPerceptron
+   lasdi.networks.CNN2D
+
+
+Module Contents
+---------------
+
+.. py:data:: act_dict
+
+.. py:class:: MultiLayerPerceptron(layer_sizes, act_type='sigmoid', reshape_index=None, reshape_shape=None, threshold=0.1, value=0.0)
+
+   Bases: :py:obj:`torch.nn.Module`
+
+
+   .. py:attribute:: n_layers
+
+
+   .. py:attribute:: layer_sizes
+
+
+   .. py:attribute:: fcs
+      :value: []
+
+
+
+   .. py:attribute:: reshape_index
+
+
+   .. py:attribute:: reshape_shape
+
+
+   .. py:attribute:: act_type
+
+
+   .. py:method:: forward(x)
+
+
+   .. py:method:: init_weight()
+
+
+.. py:class:: CNN2D(mode, channels, kernel_sizes, strides, paddings, dilations, groups=1, bias=True, padding_mode='zeros', act_type='ReLU')
+
+   Bases: :py:obj:`torch.nn.Module`
+
+
+   .. py:class:: Mode
+
+      Bases: :py:obj:`Enum`
+
+
+      Generic enumeration.
+
+      Derive from this class to define new enumerations.
+
+
+
+      .. py:attribute:: Forward
+         :value: 1
+
+
+
+      .. py:attribute:: Backward
+
+
+
+   .. py:attribute:: channels
+
+
+   .. py:attribute:: n_layers
+
+
+   .. py:attribute:: layer_sizes
+
+
+   .. py:attribute:: kernel_sizes
+
+
+   .. py:attribute:: strides
+
+
+   .. py:attribute:: paddings
+
+
+   .. py:attribute:: dilations
+
+
+   .. py:attribute:: groups
+
+
+   .. py:attribute:: bias
+
+
+   .. py:attribute:: padding_mode
+
+
+   .. py:attribute:: act
+
+
+   .. py:attribute:: fcs
+      :value: []
+
+
+
+   .. py:attribute:: batch_reshape
+      :value: None
+
+
+
+   .. py:method:: set_data_shape(data_shape: list)
+
+
+   .. py:method:: print_data_shape()
+
+
+   .. py:method:: forward(x)
+
+
+   .. py:method:: reshape_input_data(x)
+
+
+   .. py:method:: reshape_output_data(x)
+
+
+   .. py:method:: compute_input_layer_size(output_shape, kernel_size, stride, padding, dilation, mode)
+      :classmethod:
+
+
+
+   .. py:method:: compute_output_layer_size(input_shape, kernel_size, stride, padding, dilation, mode)
+      :classmethod:
+
+
+
+   .. py:method:: init_weight()
+
+

autoapi/index.html

@@ -91,6 +91,7 @@ <h1>API Reference<a class="headerlink" href="#api-reference" title="Permalink to
 </ul>
 </li>
 <li class="toctree-l2"><a class="reference internal" href="lasdi/latent_space/index.html">lasdi.latent_space</a></li>
+<li class="toctree-l2"><a class="reference internal" href="lasdi/networks/index.html">lasdi.networks</a></li>
 <li class="toctree-l2"><a class="reference internal" href="lasdi/param/index.html">lasdi.param</a></li>
 <li class="toctree-l2"><a class="reference internal" href="lasdi/physics/index.html">lasdi.physics</a><ul>
 <li class="toctree-l3"><a class="reference internal" href="lasdi/physics/burgers1d/index.html">lasdi.physics.burgers1d</a></li>

autoapi/lasdi/enums/index.html

@@ -55,6 +55,7 @@
 <li class="toctree-l4"><a class="reference internal" href="../inputs/index.html">lasdi.inputs</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../latent_dynamics/index.html">lasdi.latent_dynamics</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../latent_space/index.html">lasdi.latent_space</a></li>
+<li class="toctree-l4"><a class="reference internal" href="../networks/index.html">lasdi.networks</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../param/index.html">lasdi.param</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../physics/index.html">lasdi.physics</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../postprocess/index.html">lasdi.postprocess</a></li>

autoapi/lasdi/fd/index.html

@@ -55,6 +55,7 @@
 <li class="toctree-l4"><a class="reference internal" href="../inputs/index.html">lasdi.inputs</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../latent_dynamics/index.html">lasdi.latent_dynamics</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../latent_space/index.html">lasdi.latent_space</a></li>
+<li class="toctree-l4"><a class="reference internal" href="../networks/index.html">lasdi.networks</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../param/index.html">lasdi.param</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../physics/index.html">lasdi.physics</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../postprocess/index.html">lasdi.postprocess</a></li>

autoapi/lasdi/gp/index.html

@@ -55,6 +55,7 @@
 <li class="toctree-l4"><a class="reference internal" href="../inputs/index.html">lasdi.inputs</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../latent_dynamics/index.html">lasdi.latent_dynamics</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../latent_space/index.html">lasdi.latent_space</a></li>
+<li class="toctree-l4"><a class="reference internal" href="../networks/index.html">lasdi.networks</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../param/index.html">lasdi.param</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../physics/index.html">lasdi.physics</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../postprocess/index.html">lasdi.postprocess</a></li>

autoapi/lasdi/gplasdi/index.html

@@ -55,6 +55,7 @@
 <li class="toctree-l4"><a class="reference internal" href="../inputs/index.html">lasdi.inputs</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../latent_dynamics/index.html">lasdi.latent_dynamics</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../latent_space/index.html">lasdi.latent_space</a></li>
+<li class="toctree-l4"><a class="reference internal" href="../networks/index.html">lasdi.networks</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../param/index.html">lasdi.param</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../physics/index.html">lasdi.physics</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../postprocess/index.html">lasdi.postprocess</a></li>
@@ -262,6 +263,26 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Permalin
 <span class="sig-name descname"><span class="pre">restart_iter</span></span><em class="property"><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="pre">0</span></em><a class="headerlink" href="#lasdi.gplasdi.BayesianGLaSDI.restart_iter" title="Permalink to this definition"></a></dt>
 <dd></dd></dl>
 
+<dl class="py attribute">
+<dt class="sig sig-object py" id="lasdi.gplasdi.BayesianGLaSDI.training_loss">
+<span class="sig-name descname"><span class="pre">training_loss</span></span><em class="property"><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="pre">[]</span></em><a class="headerlink" href="#lasdi.gplasdi.BayesianGLaSDI.training_loss" title="Permalink to this definition"></a></dt>
+<dd></dd></dl>
+
+<dl class="py attribute">
+<dt class="sig sig-object py" id="lasdi.gplasdi.BayesianGLaSDI.ae_loss">
+<span class="sig-name descname"><span class="pre">ae_loss</span></span><em class="property"><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="pre">[]</span></em><a class="headerlink" href="#lasdi.gplasdi.BayesianGLaSDI.ae_loss" title="Permalink to this definition"></a></dt>
+<dd></dd></dl>
+
+<dl class="py attribute">
+<dt class="sig sig-object py" id="lasdi.gplasdi.BayesianGLaSDI.ld_loss">
+<span class="sig-name descname"><span class="pre">ld_loss</span></span><em class="property"><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="pre">[]</span></em><a class="headerlink" href="#lasdi.gplasdi.BayesianGLaSDI.ld_loss" title="Permalink to this definition"></a></dt>
+<dd></dd></dl>
+
+<dl class="py attribute">
+<dt class="sig sig-object py" id="lasdi.gplasdi.BayesianGLaSDI.coef_loss">
+<span class="sig-name descname"><span class="pre">coef_loss</span></span><em class="property"><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="pre">[]</span></em><a class="headerlink" href="#lasdi.gplasdi.BayesianGLaSDI.coef_loss" title="Permalink to this definition"></a></dt>
+<dd></dd></dl>
+
 <dl class="py method">
 <dt class="sig sig-object py" id="lasdi.gplasdi.BayesianGLaSDI.train">
 <span class="sig-name descname"><span class="pre">train</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span><a class="headerlink" href="#lasdi.gplasdi.BayesianGLaSDI.train" title="Permalink to this definition"></a></dt>

autoapi/lasdi/index.html

@@ -55,6 +55,7 @@
 <li class="toctree-l4"><a class="reference internal" href="inputs/index.html">lasdi.inputs</a></li>
 <li class="toctree-l4"><a class="reference internal" href="latent_dynamics/index.html">lasdi.latent_dynamics</a></li>
 <li class="toctree-l4"><a class="reference internal" href="latent_space/index.html">lasdi.latent_space</a></li>
+<li class="toctree-l4"><a class="reference internal" href="networks/index.html">lasdi.networks</a></li>
 <li class="toctree-l4"><a class="reference internal" href="param/index.html">lasdi.param</a></li>
 <li class="toctree-l4"><a class="reference internal" href="physics/index.html">lasdi.physics</a></li>
 <li class="toctree-l4"><a class="reference internal" href="postprocess/index.html">lasdi.postprocess</a></li>
@@ -106,6 +107,7 @@ <h2>Submodules<a class="headerlink" href="#submodules" title="Permalink to this
 <li class="toctree-l1"><a class="reference internal" href="inputs/index.html">lasdi.inputs</a></li>
 <li class="toctree-l1"><a class="reference internal" href="latent_dynamics/index.html">lasdi.latent_dynamics</a></li>
 <li class="toctree-l1"><a class="reference internal" href="latent_space/index.html">lasdi.latent_space</a></li>
+<li class="toctree-l1"><a class="reference internal" href="networks/index.html">lasdi.networks</a></li>
 <li class="toctree-l1"><a class="reference internal" href="param/index.html">lasdi.param</a></li>
 <li class="toctree-l1"><a class="reference internal" href="physics/index.html">lasdi.physics</a></li>
 <li class="toctree-l1"><a class="reference internal" href="postprocess/index.html">lasdi.postprocess</a></li>

autoapi/lasdi/inputs/index.html

@@ -55,6 +55,7 @@
 <li class="toctree-l4 current"><a class="current reference internal" href="#">lasdi.inputs</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../latent_dynamics/index.html">lasdi.latent_dynamics</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../latent_space/index.html">lasdi.latent_space</a></li>
+<li class="toctree-l4"><a class="reference internal" href="../networks/index.html">lasdi.networks</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../param/index.html">lasdi.param</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../physics/index.html">lasdi.physics</a></li>
 <li class="toctree-l4"><a class="reference internal" href="../postprocess/index.html">lasdi.postprocess</a></li>