Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Dmodel.py174 if output_leaf_index and gbdt_batch.LEAF_INDEX in predictions_dict:
175 model_fn_ops.predictions[gbdt_batch.LEAF_INDEX] = predictions_dict[
176 gbdt_batch.LEAF_INDEX]
393 if output_leaf_index and gbdt_batch.LEAF_INDEX in predictions_dict:
394 model_fn_ops.predictions[gbdt_batch.LEAF_INDEX] = predictions_dict[
395 gbdt_batch.LEAF_INDEX]
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
Dgbdt_batch.py63 LEAF_INDEX = "leaf_index" variable
115 result[LEAF_INDEX] = leaf_index