priors.ImaginaryL2

Contents

priors.ImaginaryL2#

class bartorch.priors.ImaginaryL2#

Bases: Regularizer

Squared \(\ell_2\) norm of the imaginary part (BART’s R2).

\(\tfrac{\lambda}{2} \| \operatorname{Im} x \|_2^2\).

Parameters:

  • weight (float) – \(\lambda\).

  • joint_axes (int or tuple of int, default=()) – Axes along which entries are thresholded jointly.

Also has the methods and properties of Regularizer.

Contents