|View source on GitHub|
TF Lattice premade models implement typical monotonic model architectures.
You can use TFL premade models to easily construct commonly used monotonic model
architectures. To construct a TFL premade model, construct a model configuration
tfl.configs and pass it to the premade model constructor. Note that the
inputs to the model should match the order in which they are defined in the
model_config = tfl.configs.CalibratedLatticeConfig(...) calibrated_lattice_model = tfl.premade.CalibratedLattice( model_config=model_config) calibrated_lattice_model.compile(...) calibrated_lattice_model.fit(...)
class CalibratedLattice: Premade model for Tensorflow calibrated lattice models.
class CalibratedLatticeEnsemble: Premade model for Tensorflow calibrated lattice ensemble models.
class CalibratedLinear: Premade model for Tensorflow calibrated linear models.