Searched refs:dimension_id (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/trees/ |
| D | decision_tree.cc | 56 const int32 dimension_id = split.dimension_id(); in Traverse() local
58 node_id = !sparse_feature[dimension_id].has_value() || in Traverse()
59 sparse_feature[dimension_id].get_value() <= in Traverse()
72 const int32 dimension_id = split.dimension_id(); in Traverse() local
73 node_id = sparse_feature[dimension_id].has_value() && in Traverse()
74 sparse_feature[dimension_id].get_value() <= in Traverse()
|
| /external/tensorflow/tensorflow/core/kernels/tensor_forest/ |
| D | resources.cc | 35 const int32 id, const int32 dimension_id) const { in get_prediction()
36 return decision_tree_->nodes(id).leaf().vector().value(dimension_id); in get_prediction()
|
| D | resources.h | 53 const float get_prediction(const int32 id, const int32 dimension_id) const;
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| D | custom_export_strategy.py | 169 (sorted_feature_names[feature_id], split.dimension_id))
186 (sorted_feature_names[feature_id], split.dimension_id))
231 split.dimension_id)
236 split.dimension_id)
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/ |
| D | split_handler_ops_test.py | 248 self.assertEqual(0, split_node.split.dimension_id)
271 self.assertEqual(0, split_node.split.dimension_id)
381 self.assertEqual(1, split_node.split.dimension_id)
403 self.assertEqual(2, split_node.split.dimension_id)
|
| D | prediction_ops_test.py | 96 split.dimension_id = feature_dim_id
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | split_handler_ops.cc | 686 DimensionBoundary(const int32 dimension_id, const int32 start_index) in DimensionBoundary()
687 : dimension_id(dimension_id), start_index(start_index) {} in DimensionBoundary()
689 int32 dimension_id; member
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/proto/ |
| D | tree_config.proto | 63 int32 dimension_id = 5; field
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | boosted_trees.proto | 57 int32 dimension_id = 5; field
|