Searched refs:dimension_id (Results 1 - 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/trees/
H A Ddecision_tree.cc55 const int32 dimension_id = split.dimension_id(); local
57 node_id = !sparse_feature[dimension_id].has_value() ||
58 sparse_feature[dimension_id].get_value() <=
71 const int32 dimension_id = split.dimension_id(); local
72 node_id = sparse_feature[dimension_id].has_value() &&
73 sparse_feature[dimension_id].get_value() <=
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
H A Dsplit_handler_ops.cc396 const int32 dimension_id = dimension_and_start.dimension_id; variable
457 best_dimension_idx = dimension_id;
473 best_dimension_idx = dimension_id;
512 DimensionBoundary(const int32 dimension_id, const int32 start_index)
513 : dimension_id(dimension_id), start_index(start_index) {}
515 int32 dimension_id;
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
H A Dcustom_export_strategy.py157 (sorted_feature_names[feature_id], split.dimension_id))
174 (sorted_feature_names[feature_id], split.dimension_id))
215 split.dimension_id)
220 split.dimension_id)
/external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/
H A Dsplit_handler_ops_test.py243 self.assertEqual(0, split_node.split.dimension_id)
266 self.assertEqual(0, split_node.split.dimension_id)
376 self.assertEqual(1, split_node.split.dimension_id)
398 self.assertEqual(2, split_node.split.dimension_id)
H A Dprediction_ops_test.py96 split.dimension_id = feature_dim_id

Completed in 1656 milliseconds