Commit cbf1fc8b authored by Younghee Kwon's avatar Younghee Kwon
Browse files

BoostedTreesEstimator in contrib: train_in_memory works with input_fns returning data.Dataset. 

Only one batch of data is expected, so dataset.batch() is disallowed,
and dataset.repeat() will be ignored (only the first one would be used)

PiperOrigin-RevId: 193137094
parent 91129bbb
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment