-
Notifications
You must be signed in to change notification settings - Fork 578
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
RFC: TensorFlow Lattice 2.0 #186
Conversation
@mmilanifard TFL v2.0 looks very good, and I am very excited about TFMA integrations. One minor note on the code structure. Have you considered to use tfl.layers.Linear, tfl.layers.PwlCalibrated etc instead of tfl.linear_layers.Linear, tfl.pwl_calibration_layer.PwlCalibratedLayer? The current structure is hard to memorize. |
@mmilanifard Could you post the design review notes? If this was accepted, please update this PR with the results and I'll get this merged. |
TF Lattice is launched at https://www.tensorflow.org/lattice.
Design was reviewed and accepted in Jan. Updated the RFC with the changes. Should be ready to merge. |
Launched at https://www.tensorflow.org/lattice |
Comment period is open till Jan 3, 2020.
TensorFlow Lattice 2.0
Objective
TensorFlow Lattice (TFL) is an implementation of
Deep Lattice Networks in TensorFlow. Using TFL, one can create models with guaranteed shape constraints such as monotonicity with respect to a set of features. TFL was open sourced in 2017 (https://github.com/tensorflow/lattice) and was based on TF 1.x.
This RFC covers the goals and design details of TFL 2.0 with TF 2.x eager support, Keras layers and canned estimators.