4ce2c3150094471ffa0c1a34ad9c337dfec85152,tensorflow_lattice/python/configs_test.py,,,#,125

Before Change

            "feature_b").regularizer_config_by_name("calib_hessian").l1, 0.7)


if __name__ == "__main__":
  tf.test.main()

After Change

from tensorflow_lattice.python import premade
from tensorflow_lattice.python import pwl_calibration_layer

tfl_custom_objects = {
    "CalibratedLatticeEnsemble":
        premade.CalibratedLatticeEnsemble,
    "CalibratedLattice":
        premade.CalibratedLattice,
    "CalibratedLinear":
        premade.CalibratedLinear,
    "CategoricalCalibration":
        categorical_calibration_layer.CategoricalCalibration,
    "FeatureConfig":
        configs.FeatureConfig,
    "RegularizerConfig":
        configs.RegularizerConfig,
    "TrustConfig":
        configs.TrustConfig,
    "DominanceConfig":
        configs.DominanceConfig,
    "CalibratedLatticeEnsembleConfig":
        configs.CalibratedLatticeEnsembleConfig,
    "CalibratedLatticeConfig":
        configs.CalibratedLatticeConfig,
    "CalibratedLinearConfig":
        configs.CalibratedLinearConfig,
    "Lattice":
        lattice_layer.Lattice,
    "Linear":
        linear_layer.Linear,
    "PWLCalibration":
        pwl_calibration_layer.PWLCalibration,
}


class ConfigsTest(tf.test.TestCase):

Instances