pararealml.operators.ml package

Submodules

pararealml.operators.ml.deeponet module

class pararealml.operators.ml.deeponet.DeepONet(branch_layer_sizes: Sequence[int], trunk_layer_sizes: Sequence[int], output_size: int, branch_initialization: str = 'glorot_uniform', trunk_initialization: str = 'glorot_uniform', branch_activation: Optional[str] = 'tanh', trunk_activation: Optional[str] = 'tanh')[source]

Bases: keras.engine.training.Model

A Deep Operator Network model.

See: https://arxiv.org/abs/1910.03193

call[source]

pararealml.operators.ml.fnn_regressor module

class pararealml.operators.ml.fnn_regressor.FNNRegressor(layer_sizes: Sequence[int], initialization: str = 'glorot_uniform', activation: Optional[str] = 'tanh')[source]

Bases: keras.engine.sequential.Sequential

A fully-connected feedforward neural network regression model.

Module contents