2fd542242f2dcf1820b616461d2801102c36957e,tensorflow_lattice/python/estimators/hparams.py,CalibratedEtlHParams,__init__,#CalibratedEtlHParams#Any#,592

Before Change

    self.set_param_type("linear_embedding_calibration_max", float)
    self.set_param_type("linear_embedding_calibration_num_keypoints", int)

    self.set_param_type("lattice_l1_reg", float)
    self.set_param_type("lattice_l2_reg", float)
    self.set_param_type("lattice_l1_torsion_reg", float)
    self.set_param_type("lattice_l2_torsion_reg", float)
    self.set_param_type("lattice_l1_laplacian_reg", float)
    self.set_param_type("lattice_l2_laplacian_reg", float)

After Change

        "linear_embedding_calibration_max": 100.0,
        "linear_embedding_calibration_num_keypoints": 100,
    }
    regularizer_hparam_names = [
        "lattice_{}".format(regularizer_name)
        for regularizer_name in regularizers.LATTICE_REGULARIZERS
    ]
    args.update({
        regularizer_name: None for regularizer_name in regularizer_hparam_names
    })
    args.update(kwargs)
    super(CalibratedEtlHParams, self).__init__(feature_names, **args)
    self.set_param_type("monotonic_lattice_rank", int)
    self.set_param_type("monotonic_num_lattices", int)
    self.set_param_type("monotonic_lattice_size", int)
    self.set_param_type("non_monotonic_lattice_rank", int)
    self.set_param_type("non_monotonic_num_lattices", int)
    self.set_param_type("non_monotonic_lattice_size", int)
    self.set_param_type("linear_embedding_calibration_min", float)
    self.set_param_type("linear_embedding_calibration_max", float)
    self.set_param_type("linear_embedding_calibration_num_keypoints", int)
    for regularizer_name in regularizer_hparam_names:
      self.set_param_type(regularizer_name, float)

Instances