Attend the Women in ML Symposium on December 7 Register now

Module: tfl.model_info

Stay organized with collections Save and categorize content based on your preferences.

Classes defining trained TFL model structure and parameter information.

This package provides representations and tools for analysis of a trained TF Lattice model, e.g. a canned estimator in saved model format.


class CategoricalCalibrationNode: Represetns a categorical calibration layer.

class InputFeatureNode: Input features to the model.

class KroneckerFactoredLatticeNode: Represents a kronecker-factored lattice layer.

class LatticeNode: Represetns a lattice layer.

class LinearNode: Represents a linear layer.

class MeanNode: Represents an averaging layer.

class ModelGraph: Model info and parameter as a graph.

class PWLCalibrationNode: Represetns a PWL calibration layer.

absolute_import Instance of __future__._Feature
division Instance of __future__._Feature
print_function Instance of __future__._Feature