boax.core.models.means.linear

Contents

boax.core.models.means.linear#

boax.core.models.means.linear(scale, bias)#

The linear mean function.

Computes y = f(x) = scale * x + bias.

Example

>>> mean = linear(jnp.array([0.2, 3.0]), 5.0)
>>> ys = mean(xs)
Parameters:
  • scale (Union[Array, ndarray, bool, number]) – The d-dim scale parameter.

  • bias (Union[Array, ndarray, bool, number, float, int]) – The numeric bias.

Return type:

Mean

Returns:

A linear mean function.