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

Question

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

fuyanlu1989 opened this issue 5 years ago · comments

fuyanlu1989 commented 5 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 · Answer 1 · Tue Oct 01 2019 03:17:17 GMT+0800 (China Standard Time)

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 · Answer 2 · Tue Oct 01 2019 03:33:50 GMT+0800 (China Standard Time)

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