Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/models/
Dmultiple_additive_trees.cc53 const float tree_weight = config.tree_weights(tree_idx); in Predict() local
68 const float value = tree_weight * leaf.value(logit_dim); in Predict()
76 const float value = tree_weight * leaf.value(i); in Predict()
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Dcustom_export_strategy.py131 tree_weight = dtec.tree_weights[tree_idx]
145 new_value.float_value = weight * tree_weight
150 new_value.float_value = weight * tree_weight
/external/tensorflow/tensorflow/contrib/tensor_forest/python/
Dtensor_forest_test.py128 tree_weight = {'decisionTree':
144 restored_tree_param = ParseDict(tree_weight,
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dprediction_ops.cc236 const float tree_weight = resource->GetTreeWeight(tree_id); in Compute() local
240 tree_logits[j] += tree_weight * leaf_logits[j]; in Compute()