Home
last modified time | relevance | path

Searched refs:ensemble_stats (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
Dgbdt_batch.py87 ensemble_stats, argument
110 result[NUM_LAYERS_ATTEMPTED] = ensemble_stats.attempted_layers
111 result[NUM_TREES_ATTEMPTED] = ensemble_stats.attempted_trees
271 def _dropout_params(mode, ensemble_stats): argument
284 seed = ensemble_stats.attempted_trees
475 ensemble_stats = training_ops.tree_ensemble_stats(ensemble_handle,
486 apply_dropout, seed = _dropout_params(mode, ensemble_stats)
489 with ops.control_dependencies(ensemble_stats):
539 ensemble_stats, used_handlers, leaf_index)