Skip to content
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

Merged
merged 5 commits into from
Apr 15, 2020
Merged

Conversation

mmilanifard
Copy link
Contributor

@mmilanifard mmilanifard commented Dec 6, 2019

Comment period is open till Jan 3, 2020.

TensorFlow Lattice 2.0

Status Proposed
RFC # 186
Author(s) Mahdi Milani Fard (mmilanifard@google.com), Oleksandr Mangylov (amangy@google.com)
Sponsor Zhenyu Tan (tanzheny@google.com), Karmel Allison (karmel@google.com)
Updated 2019-12-06

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.

@si-you
Copy link

si-you commented Jan 1, 2020

@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.

@ematejska
Copy link
Contributor

@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.

@mmilanifard
Copy link
Contributor Author

Design was reviewed and accepted in Jan. Updated the RFC with the changes. Should be ready to merge.

@mmilanifard
Copy link
Contributor Author

Launched at https://www.tensorflow.org/lattice

@ematejska ematejska added RFC: Accepted RFC Design Document: Accepted by Review and removed RFC: Proposed RFC Design Document labels Apr 15, 2020
@ematejska ematejska merged commit 1d588c4 into tensorflow:master Apr 15, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cla: yes RFC: Accepted RFC Design Document: Accepted by Review
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants