search

tensorflow / lattice

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

Lattice for regression #42

Closed AishwaryaSivaraman closed 4 years ago

AishwaryaSivaraman commented 4 years ago

Do you have an example of running lattice for a regression dataset?

mmilanifard commented 4 years ago

There are regressor training examples on the Boston housing dataset here: https://github.com/tensorflow/lattice/blob/master/tensorflow_lattice/python/estimators_test.py