Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/boosted_trees/resources/
Ddecision_tree_ensemble_resource.h65 boosted_trees::trees::GrowingMetadata* const growing_metadata = in IncrementAttempts() local
67 growing_metadata->set_num_layers_attempted( in IncrementAttempts()
68 growing_metadata->num_layers_attempted() + 1); in IncrementAttempts()
71 growing_metadata->set_num_trees_attempted( in IncrementAttempts()
72 growing_metadata->num_trees_attempted() + 1); in IncrementAttempts()
136 decision_tree_ensemble_->growing_metadata().used_handler_ids().size()); in GetUsedHandlers()
138 decision_tree_ensemble_->growing_metadata().used_handler_ids()) { in GetUsedHandlers()
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dresources.cc150 tree_ensemble_->growing_metadata().last_layer_node_start(); in GetLastLayerNodesRange()
151 *node_range_end = tree_ensemble_->growing_metadata().last_layer_node_end(); in GetLastLayerNodesRange()
160 return tree_ensemble_->growing_metadata().num_layers_attempted(); in GetNumLayersAttempted()
237 tree_ensemble_->growing_metadata().num_layers_attempted() + 1); in UpdateGrowingMetadata()
246 tree_ensemble_->growing_metadata().num_trees_attempted() + 1); in UpdateGrowingMetadata()
Dboosted_trees.proto153 GrowingMetadata growing_metadata = 4; field
/external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/
Dmodel_ops_test.py316 tree_ensemble_config.growing_metadata.used_handler_ids.append(1)
317 tree_ensemble_config.growing_metadata.used_handler_ids.append(5)
Dprediction_ops_test.py905 tree_ensemble_config.growing_metadata.num_trees_attempted = num_trees
906 tree_ensemble_config.growing_metadata.num_layers_attempted = num_trees
Dtraining_ops_test.py2574 [0, 2], tree_ensemble_config.growing_metadata.used_handler_ids)
/external/tensorflow/tensorflow/contrib/boosted_trees/proto/
Dtree_config.proto171 GrowingMetadata growing_metadata = 4; field
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dtraining_ops.cc944 ensemble_config.growing_metadata().num_trees_attempted(); in Compute()
971 ensemble_config.growing_metadata().num_layers_attempted(); in Compute()
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
Dgbdt_batch_test.py1793 self.assertEquals(output.growing_metadata.num_layers_attempted, 1)
1809 self.assertEquals(output.growing_metadata.num_layers_attempted, 2)