Commit d48c55db authored by Younghee Kwon's avatar Younghee Kwon Committed by TensorFlower Gardener
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 9b24fb8d
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