tensorflow / lattice

Lattice methods in TensorFlow
Apache License 2.0
518 stars 93 forks source link

calibrated linear model bug fix and update tensorflow branch #13

Closed si-you closed 6 years ago

si-you commented 6 years ago

I found a bug when training a monotonic calibrated linear model. When we set a monotonicity for a calibrated linear model without any bound_min and bound_max, a calibration layer raises an error because it tries to enforce non-existing bound_min and bound_max to missing output value. A test added in this pull request can reproduce this error.