| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/models/ |
| D | multiple_additive_trees_test.cc | 43 MultipleAdditiveTreesTest() : batch_features_(2) { in MultipleAdditiveTreesTest()
49 batch_features_.Initialize({dense_matrix}, {}, {}, {}, {}, {}, {})); in MultipleAdditiveTreesTest()
52 boosted_trees::utils::BatchFeatures batch_features_; member in tensorflow::boosted_trees::models::__anon130348f60111::MultipleAdditiveTreesTest
64 MultipleAdditiveTrees::Predict(tree_ensemble_config, {}, batch_features_, in TEST_F()
102 batch_features_, &threads, output_matrix); in TEST_F()
111 MultipleAdditiveTrees::Predict(weighted, {0, 1}, batch_features_, &threads, in TEST_F()
120 MultipleAdditiveTrees::Predict(tree_ensemble_config, {1}, batch_features_, in TEST_F()
127 MultipleAdditiveTrees::Predict(tree_ensemble_config, {0}, batch_features_, in TEST_F()
134 MultipleAdditiveTrees::Predict(tree_ensemble_config, {}, batch_features_, in TEST_F()
175 batch_features_, &threads, output_matrix); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/partitioners/ |
| D | example_partitioner_test.cc | 29 batch_features_(2) { in ExamplePartitionerTest()
32 batch_features_.Initialize({dense_matrix_}, {}, {}, {}, {}, {}, {})); in ExamplePartitionerTest()
37 boosted_trees::utils::BatchFeatures batch_features_; member in tensorflow::boosted_trees::learner::__anond206f42b0111::ExamplePartitionerTest
43 ExamplePartitioner::UpdatePartitions(tree_config, batch_features_, 1, in TEST_F()
66 ExamplePartitioner::UpdatePartitions(tree_config, batch_features_, 1, in TEST_F()
89 ExamplePartitioner::PartitionExamples(tree_config, batch_features_, 1, in TEST_F()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/trees/ |
| D | decision_tree_test.cc | 28 DecisionTreeTest() : batch_features_(2) { in DecisionTreeTest()
56 TF_EXPECT_OK(batch_features_.Initialize( in DecisionTreeTest()
91 utils::BatchFeatures batch_features_; member in tensorflow::boosted_trees::trees::__anonce9f1edb0111::DecisionTreeTest
96 auto example = (*batch_features_.examples_iterable(0, 1).begin()); in TEST_F()
103 auto example = (*batch_features_.examples_iterable(0, 1).begin()); in TEST_F()
110 auto example = (*batch_features_.examples_iterable(0, 1).begin()); in TEST_F()
124 auto example_iterable = batch_features_.examples_iterable(0, 2); in TEST_F()
135 auto example_iterable = batch_features_.examples_iterable(0, 2); in TEST_F()
231 auto example_iterable = batch_features_.examples_iterable(0, 2); in TEST_F()
251 auto example_iterable = batch_features_.examples_iterable(0, 2); in TEST_F()
[all …]
|