xnn.common.train.losses

Weighted energy / force / stress (and dipole / polarizability / charge) loss.

Functions

weighted_loss(pred, data, energy_weight, ...)

Compute a weighted sum of energy, force and stress loss terms.