cayleypy.train.make_loss

cayleypy.train.make_loss(name: str, tau: float = 0.5) → Loss[source]

Creates loss by name.

Parameters:
  • name – Name of the loss - either “mse” or “pinball”.

  • tau – Quantile for the pinball loss (ignored by other losses).

Returns:

The loss.