More on calibration using TF-Lattice here 4. The idea is that once you trained the TF-DF model, one connects its output to tfl.layers.PWLCalibration 3 with monotonicity=1 (increasing). And retrain this composed model (the TF-DF model will be frozen, only the calibration will be updated).
Hopefully we’ll add automatic calibration in the near future.