Skip to content

Model Types

The PyTorch Lattice library currently supports two types of calibrated modeling:

  • CalibratedLinear: a calibrated linear model combines calibrated features using a standard linear layer, optionally followed by an output calibrator.

  • CalibratedLattice: a calibrated lattice model combines calibrated features using a lattice layer, optionally followed by an output calibrator. The lattice layer can learn higher-order feature interactions, which can help increase model flexibility and thereby performance on more complex prediction tasks.