pymor.models.neural_network¶
Remark on the documentation:
Due to an issue in autoapi, the classes NeuralNetworkStatefreeOutputModel,
NeuralNetworkInstationaryModel, NeuralNetworkInstationaryStatefreeOutputModel
and FullyConnectedNN do not appear in the documentation,
see https://github.com/pymor/pymor/issues/1343.
Module Contents¶
Classes¶
Class for models of stationary problems that use artificial neural networks. |
- class pymor.models.neural_network.NeuralNetworkModel(neural_network, parameters={}, output_functional=None, products=None, error_estimator=None, visualizer=None, name=None)[source]¶
Bases:
pymor.models.interface.ModelClass for models of stationary problems that use artificial neural networks.
This class implements a
Modelthat uses a neural network for solving.Parameters
- neural_network
The neural network that approximates the mapping from parameter space to solution space. Should be an instance of
FullyConnectedNNwith input size that matches the (total) number of parameters and output size equal to the dimension of the reduced space.- parameters
Parametersof the reduced order model (the same as used in the full-order model).- output_functional
Operatormapping a given solution to the model output. In many applications, this will be aFunctional, i.e. anOperatormapping to scalars. This is not required, however.- products
A dict of inner product
Operatorsdefined on the discrete space the problem is posed on. For each product with key'x'a corresponding attributex_product, as well as a norm methodx_normis added to the model.- error_estimator
An error estimator for the problem. This can be any object with an
estimate_error(U, mu, m)method. Iferror_estimatoris notNone, anestimate_error(U, mu)method is added to the model which will callerror_estimator.estimate_error(U, mu, self).- visualizer
A visualizer for the problem. This can be any object with a
visualize(U, m, ...)method. Ifvisualizeris notNone, avisualize(U, *args, **kwargs)method is added to the model which forwards its arguments to the visualizer’svisualizemethod.- name
Name of the model.
- _compute_solution(self, mu=None, **kwargs)[source]¶
Compute the model’s solution for
parameter valuesmu.This method is called by the default implementation of
computeinpymor.models.interface.Model.Parameters
- mu
Parameter valuesfor which to compute the solution.- kwargs
Additional keyword arguments to customize how the solution is computed or to select additional data to be returned.
Returns
VectorArraywith the computed solution or a dict which at least must contain the key'solution'.