Home
last modified time | relevance | path

Searched refs:dropped_trees (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
Ddropout_utils_test.cc55 std::vector<int32> dropped_trees; in TEST_F() local
66 weights_, &dropped_trees, in TEST_F()
70 EXPECT_TRUE(dropped_trees.empty()); in TEST_F()
80 weights_, &dropped_trees, in TEST_F()
84 EXPECT_EQ(kNumTrees, dropped_trees.size()); in TEST_F()
102 weights_, &dropped_trees, in TEST_F()
106 EXPECT_NEAR(500, kNumTrees - dropped_trees.size(), 100); in TEST_F()
107 total_num_trees += kNumTrees - dropped_trees.size(); in TEST_F()
111 for (const auto& tree : dropped_trees) { in TEST_F()
114 EXPECT_EQ(ids.size(), dropped_trees.size()); in TEST_F()
[all …]
Ddropout_utils.cc38 const std::vector<float>& weights, std::vector<int32>* dropped_trees, in DropOutTrees() argument
41 if (dropped_trees == nullptr) { in DropOutTrees()
61 dropped_trees->clear(); in DropOutTrees()
95 dropped_trees->push_back(i); in DropOutTrees()
100 std::sort(dropped_trees->begin(), dropped_trees->end()); in DropOutTrees()
101 for (const int32 dropped_tree : *dropped_trees) { in DropOutTrees()
109 const std::vector<int32>& dropped_trees, in GetTreesWeightsForAddingTrees() argument
120 const int num_dropped = dropped_trees.size(); in GetTreesWeightsForAddingTrees()
139 for (int32 i = 0; i < dropped_trees.size(); ++i) { in GetTreesWeightsForAddingTrees()
140 const int32 dropped = dropped_trees[i]; in GetTreesWeightsForAddingTrees()
Ddropout_utils.h45 const std::vector<float>& weights, std::vector<int32>* dropped_trees,
63 const std::vector<int32>& dropped_trees,
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dprediction_ops.cc236 std::vector<int32> dropped_trees; in DoCompute() local
260 weights, &dropped_trees, &original_weights)); in DoCompute()
266 ensemble_resource->decision_tree_ensemble(), dropped_trees, in DoCompute()
320 {2, static_cast<int64>(dropped_trees.size())}, in DoCompute()
323 for (int32 i = 0; i < dropped_trees.size(); ++i) { in DoCompute()
324 output_dropout_info(0, i) = dropped_trees[i]; in DoCompute()
Dtraining_ops.cc585 std::vector<int32> dropped_trees; in UpdateTreeWeightsIfDropout() local
589 &dropped_trees, &dropped_trees_weights); in UpdateTreeWeightsIfDropout()
596 if (!dropped_trees.empty()) { in UpdateTreeWeightsIfDropout()
599 dropped_trees, dropped_trees_weights, current_tree, in UpdateTreeWeightsIfDropout()