Constant model#

class fortuna.model.constant.ConstantModel(output_dim, initializer_fun=<function zeros>, parent=<flax.linen.module._Sentinel object>, name=None)[source]#

Bases: Module

A constant model, that is \(f(\theta, x) = \theta\).

Parameters:

  • output_dim (int) – The output model dimension.

  • initializer_fun (Optional[Initializer]) – Function to initialize the model parameters. This must be one of the available options in flax.linen.initializers.