torchref.refinement.targets.geometry.angles module

class torchref.refinement.targets.geometry.angles.AngleTarget(model=None, verbose=0)[source]

Bases: GeometryTarget

Angle restraint target (Gaussian NLL).

NLL = 0.5 * ((θ - θ₀) / σ)² + log(σ) + 0.5 * log(2π)

name: str = 'geometry/angle'
__init__(model=None, verbose=0)[source]

Initialize model target.

Parameters:
  • model (Model, optional) – Reference to the Model object (optional for empty init).

  • verbose (int, optional) – Verbosity level. Default is 0.

forward()[source]

Compute and return the loss. Override in subclasses.

stats()[source]

Get angle restraint statistics.