search

tensorflow / lattice

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

How to add sample weights to tensorflow_lattice.calibrated_etl_classifier and other estimators #39

Closed fuyanlu1989 closed 4 years ago

fuyanlu1989 commented 4 years ago

In tensorflow.estimator.DNNRegressor, we can add sample weights by argument weight_column. But we didn't find this for tensorflow lattice estimators. Is it a missing feature? Could you please add this feature?

mmilanifard commented 4 years ago

I'm not sure if I understand your question. calibrated_etl_regressor has a weight_column argument. Same for all other estimators.

https://github.com/tensorflow/lattice/blob/master/tensorflow_lattice/python/estimators/calibrated_etl.py#L568

fuyanlu1989 commented 4 years ago

Thanks! I didn't find it in examples. And not found instruction online. Not dig into the source code.