Commit 9db84049 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

boosted_trees: some cleanups. 

 - removed learner_config.num_classes in training_ops which is unnecessary.
 - replaced num_classes in gbdt_batch.py with logits_dimension where possible.
 - simplified prediction_ops_test.

PiperOrigin-RevId: 174399706
parent 6ace5e04
Expand all 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