HomeSort by relevance Sort by last modified time
    Searched refs:RandomTreeGen (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/
random_tree_gen.h 30 class RandomTreeGen {
32 RandomTreeGen(tensorflow::random::SimplePhilox* rng, int dense_feature_size,
68 TF_DISALLOW_COPY_AND_ASSIGN(RandomTreeGen);
random_tree_gen.cc 59 RandomTreeGen::RandomTreeGen(tensorflow::random::SimplePhilox* rng,
78 RandomTreeGen::GenerateEnsemble(int depth, int tree_count) {
89 DecisionTreeConfig RandomTreeGen::Generate(const DecisionTreeConfig& tree) {
107 DecisionTreeConfig RandomTreeGen::Generate(int depth) {
130 void RandomTreeGen::Combine(DecisionTreeConfig* root,
151 void RandomTreeGen::ShiftNodeIndex(DecisionTreeConfig* tree, int shift) {
178 void RandomTreeGen::GenerateSplit(TreeNode* node, int left_id, int right_id) {
random_tree_gen_main.cc 56 tensorflow::boosted_trees::testutil::RandomTreeGen tree_gen(

Completed in 282 milliseconds