Home
last modified time | relevance | path

Searched refs:boosted_trees (Results 1 – 25 of 153) sorted by relevance

1234567

/external/tensorflow/tensorflow/contrib/boosted_trees/
DBUILD94 "//tensorflow/contrib/boosted_trees:batch_ops_utils_py",
95 "//tensorflow/contrib/boosted_trees:boosted_trees_ops_py",
96 "//tensorflow/contrib/boosted_trees/lib:categorical_split_handler",
97 "//tensorflow/contrib/boosted_trees/lib:ordinal_split_handler",
98 "//tensorflow/contrib/boosted_trees/proto:learner_proto_py",
132 "//tensorflow/contrib/boosted_trees/proto:learner_proto_py",
133 "//tensorflow/contrib/boosted_trees/proto:tree_config_proto_py",
157 "//tensorflow/contrib/boosted_trees/proto:learner_proto_py",
158 "//tensorflow/contrib/boosted_trees/proto:tree_config_proto_py",
177 "//tensorflow/contrib/boosted_trees/proto:learner_proto_py",
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/
Drandom_tree_gen.h25 namespace boosted_trees {
38 boosted_trees::trees::DecisionTreeConfig Generate(int depth);
42 boosted_trees::trees::DecisionTreeConfig Generate(
43 const boosted_trees::trees::DecisionTreeConfig& tree);
46 boosted_trees::trees::DecisionTreeEnsembleConfig GenerateEnsemble(
55 void Combine(boosted_trees::trees::DecisionTreeConfig* root,
56 boosted_trees::trees::DecisionTreeConfig* left_branch,
57 boosted_trees::trees::DecisionTreeConfig* right_branch);
61 void ShiftNodeIndex(boosted_trees::trees::DecisionTreeConfig* tree,
65 void GenerateSplit(boosted_trees::trees::TreeNode* node, int left_id,
Drandom_tree_gen.cc22 namespace boosted_trees { namespace
25 using boosted_trees::trees::DenseFloatBinarySplit;
26 using tensorflow::boosted_trees::trees::DecisionTreeConfig;
27 using tensorflow::boosted_trees::trees::TreeNode;
67 boosted_trees::trees::DecisionTreeEnsembleConfig* ret) { in AddWeightAndMetadata()
77 boosted_trees::trees::DecisionTreeEnsembleConfig
79 boosted_trees::trees::DecisionTreeEnsembleConfig ret; in GenerateEnsemble()
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dtraining_ops.cc26 using tensorflow::boosted_trees::learner::LearningRateDropoutDrivenConfig;
28 namespace boosted_trees { namespace
32 using boosted_trees::learner::LearnerConfig;
33 using boosted_trees::learner::LearningRateConfig;
34 using boosted_trees::trees::Leaf;
35 using boosted_trees::trees::TreeNode;
36 using boosted_trees::trees::TreeNodeMetadata;
37 using boosted_trees::utils::DropoutUtils;
62 const boosted_trees::learner::LearnerConfig& learner_config, in UpdateBestSplit()
75 boosted_trees::learner::LearnerConfig::PRE_PRUNE && in UpdateBestSplit()
[all …]
Dprediction_ops.cc41 using tensorflow::boosted_trees::learner::AveragingConfig;
42 using tensorflow::boosted_trees::trees::DecisionTreeEnsembleConfig;
45 namespace boosted_trees { namespace
47 using boosted_trees::learner::LearnerConfig;
48 using boosted_trees::learner::LearningRateConfig;
49 using boosted_trees::learner::LearningRateDropoutDrivenConfig;
50 using boosted_trees::models::DecisionTreeEnsembleResource;
51 using boosted_trees::models::MultipleAdditiveTrees;
52 using boosted_trees::utils::DropoutUtils;
53 using boosted_trees::utils::TensorUtils;
[all …]
Dmodel_ops.cc27 namespace boosted_trees { namespace
29 using boosted_trees::models::DecisionTreeEnsembleResource;
47 auto* result = new boosted_trees::models::DecisionTreeEnsembleResource(); in Compute()
72 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource; in Compute()
91 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource; in Compute()
115 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource; in Compute()
149 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource; in Compute()
199 IsResourceInitialized<boosted_trees::models::DecisionTreeEnsembleResource>);
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/
DBUILD2 # This directory contains common utilities used in boosted_trees.
9 "//tensorflow/contrib/boosted_trees:__subpackages__",
10 "//tensorflow/contrib/boosted_trees:friends",
43 "//tensorflow/contrib/boosted_trees/proto:learner_proto_cc",
105 "//tensorflow/contrib/boosted_trees/proto:tree_config_proto_cc",
122 "//tensorflow/contrib/boosted_trees/proto:tree_config_proto_cc",
135 "//tensorflow/contrib/boosted_trees/resources:decision_tree_ensemble_resource",
163 "//tensorflow/contrib/boosted_trees/proto:tree_config_proto_cc",
237 "//tensorflow/contrib/boosted_trees/lib:utils",
238 "//tensorflow/contrib/boosted_trees/proto:tree_config_proto_cc",
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/partitioners/
Dexample_partitioner.cc19 namespace boosted_trees { namespace
23 const boosted_trees::trees::DecisionTreeConfig& tree_config, in UpdatePartitions()
24 const boosted_trees::utils::BatchFeatures& features, in UpdatePartitions()
41 example_partition = boosted_trees::trees::DecisionTree::Traverse( in UpdatePartitions()
51 boosted_trees::utils::ParallelFor(batch_size, desired_parallelism, in UpdatePartitions()
56 const boosted_trees::trees::DecisionTreeConfig& tree_config, in PartitionExamples()
57 const boosted_trees::utils::BatchFeatures& features, in PartitionExamples()
73 uint32 partition = boosted_trees::trees::DecisionTree::Traverse( in PartitionExamples()
84 boosted_trees::utils::ParallelFor(batch_size, desired_parallelism, in PartitionExamples()
Dexample_partitioner_test.cc21 namespace boosted_trees { namespace
37 boosted_trees::utils::BatchFeatures batch_features_;
41 boosted_trees::trees::DecisionTreeConfig tree_config; in TEST_F()
53 boosted_trees::trees::DecisionTreeConfig tree_config; in TEST_F()
76 boosted_trees::trees::DecisionTreeConfig tree_config; in TEST_F()
/external/tensorflow/tensorflow/contrib/boosted_trees/proto/
Dsplit_info.proto5 package tensorflow.boosted_trees.learner;
7 import "tensorflow/contrib/boosted_trees/proto/tree_config.proto";
12 tensorflow.boosted_trees.trees.TreeNode split_node = 1;
15 tensorflow.boosted_trees.trees.Leaf left_child = 2;
18 tensorflow.boosted_trees.trees.Leaf right_child = 3;
22 tensorflow.boosted_trees.trees.TreeNode split_node = 1;
23 repeated tensorflow.boosted_trees.trees.Leaf children = 2;
/external/tensorflow/tensorflow/contrib/boosted_trees/resources/
Ddecision_tree_ensemble_resource.h25 namespace boosted_trees {
35 boosted_trees::trees::DecisionTreeEnsembleConfig>(&arena_)) {} in DecisionTreeEnsembleResource()
42 const boosted_trees::trees::DecisionTreeEnsembleConfig&
65 boosted_trees::trees::GrowingMetadata* const growing_metadata = in IncrementAttempts()
76 boosted_trees::trees::DecisionTreeConfig* AddNewTree(const float weight) { in AddNewTree()
79 boosted_trees::trees::DecisionTreeMetadata* const metadata = in AddNewTree()
92 boosted_trees::trees::DecisionTreeConfig* LastTree() { in LastTree()
98 boosted_trees::trees::DecisionTreeMetadata* LastTreeMetadata() { in LastTreeMetadata()
168 boosted_trees::trees::DecisionTreeEnsembleConfig>(&arena_); in Reset()
176 boosted_trees::trees::DecisionTreeEnsembleConfig* decision_tree_ensemble_;
DBUILD7 "//tensorflow/contrib/boosted_trees:__subpackages__",
8 "//tensorflow/contrib/boosted_trees:friends",
26 "//tensorflow/contrib/boosted_trees/lib:weighted_quantiles",
27 "//tensorflow/contrib/boosted_trees/proto:quantiles_proto_cc",
37 "//tensorflow/contrib/boosted_trees/lib:trees",
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
DBUILD34 "//tensorflow/contrib/boosted_trees:gbdt_batch",
35 "//tensorflow/contrib/boosted_trees:model_ops_py",
104 "//tensorflow/contrib/boosted_trees:gbdt_batch",
105 "//tensorflow/contrib/boosted_trees/proto:tree_config_proto_py",
124 "//tensorflow/contrib/boosted_trees/proto:tree_config_proto_py",
138 "//tensorflow/contrib/boosted_trees:losses",
153 "//tensorflow/contrib/boosted_trees:gbdt_batch",
154 "//tensorflow/contrib/boosted_trees:model_ops_py",
187 "//tensorflow/contrib/boosted_trees:gbdt_batch",
207 "//tensorflow/contrib/boosted_trees:gbdt_batch",
/external/tensorflow/tensorflow/contrib/cmake/
Dtf_core_kernels.cmake45 "${tensorflow_source_dir}/tensorflow/contrib/boosted_trees/kernels/model_ops.cc"
46 "${tensorflow_source_dir}/tensorflow/contrib/boosted_trees/kernels/prediction_ops.cc"
47 "${tensorflow_source_dir}/tensorflow/contrib/boosted_trees/kernels/quantile_ops.cc"
48 "${tensorflow_source_dir}/tensorflow/contrib/boosted_trees/kernels/split_handler_ops.cc"
49 "${tensorflow_source_dir}/tensorflow/contrib/boosted_trees/kernels/stats_accumulator_ops.cc"
50 "${tensorflow_source_dir}/tensorflow/contrib/boosted_trees/kernels/training_ops.cc"
51 "${tensorflow_source_dir}/tensorflow/contrib/boosted_trees/lib/utils/batch_features.cc"
52 "${tensorflow_source_dir}/tensorflow/contrib/boosted_trees/lib/utils/dropout_utils.cc"
53 "${tensorflow_source_dir}/tensorflow/contrib/boosted_trees/lib/utils/examples_iterable.cc"
54 "${tensorflow_source_dir}/tensorflow/contrib/boosted_trees/lib/utils/parallel_for.cc"
[all …]
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dresources.cc27 protobuf::Arena::CreateMessage<boosted_trees::TreeEnsemble>( in BoostedTreesEnsembleResource()
61 case boosted_trees::Node::kBucketizedSplit: { in next_node()
68 case boosted_trees::Node::kCategoricalSplit: { in next_node()
86 if (node.node_case() == boosted_trees::Node::kLeaf) { in node_value()
122 DCHECK(node->node_case() == boosted_trees::Node::kLeaf); in set_node_value()
168 return node.node_case() == boosted_trees::Node::kLeaf; in is_leaf()
174 DCHECK_EQ(node.node_case(), boosted_trees::Node::kBucketizedSplit); in feature_id()
181 DCHECK_EQ(node.node_case(), boosted_trees::Node::kBucketizedSplit); in bucket_threshold()
188 DCHECK_EQ(node.node_case(), boosted_trees::Node::kBucketizedSplit); in left_id()
195 DCHECK_EQ(node.node_case(), boosted_trees::Node::kBucketizedSplit); in right_id()
[all …]
DBUILD22 "boosted_trees.proto",
37 "//tensorflow/core/kernels/boosted_trees:boosted_trees_proto_cc",
48 "//tensorflow/core/kernels/boosted_trees:boosted_trees_proto_cc",
64 "//tensorflow/core/kernels/boosted_trees:boosted_trees_proto_cc",
86 "//tensorflow/core/kernels/boosted_trees:boosted_trees_proto_cc",
96 "//tensorflow/core/kernels/boosted_trees/quantiles:weighted_quantiles",
/external/tensorflow/tensorflow/core/kernels/tensor_forest/
Dresources.cc22 const boosted_trees::Tree& TensorForestTreeResource::decision_tree() const { in decision_tree()
32 protobuf::Arena::CreateMessage<boosted_trees::Tree>(&arena_)) {} in TensorForestTreeResource()
42 using boosted_trees::Node; in TraverseTree()
43 using boosted_trees::Tree; in TraverseTree()
68 decision_tree_ = protobuf::Arena::CreateMessage<boosted_trees::Tree>(&arena_); in Reset()
/external/tensorflow/tensorflow/python/estimator/canned/
Dboosted_trees.py26 from tensorflow_estimator.python.estimator.canned import boosted_trees
30 boosted_trees.__all__ = [
31 s for s in dir(boosted_trees) if not s.startswith('__')
34 from tensorflow_estimator.python.estimator.canned.boosted_trees import *
/external/tensorflow/tensorflow/contrib/estimator/python/estimator/
Dboosted_trees.py26 from tensorflow_estimator.contrib.estimator.python.estimator import boosted_trees
30 boosted_trees.__all__ = [
31 s for s in dir(boosted_trees) if not s.startswith('__')
34 from tensorflow_estimator.contrib.estimator.python.estimator.boosted_trees import *
/external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/
Dprediction_ops.py21 from tensorflow.contrib.boosted_trees.python.ops import boosted_trees_ops_loader
22 from tensorflow.contrib.boosted_trees.python.ops.gen_prediction_ops import gradient_trees_partition…
23 from tensorflow.contrib.boosted_trees.python.ops.gen_prediction_ops import gradient_trees_prediction
24 from tensorflow.contrib.boosted_trees.python.ops.gen_prediction_ops import gradient_trees_predictio…
Dmodel_ops.py23 from tensorflow.contrib.boosted_trees.python.ops import boosted_trees_ops_loader
25 from tensorflow.contrib.boosted_trees.python.ops import gen_model_ops
26 from tensorflow.contrib.boosted_trees.python.ops.gen_model_ops import tree_ensemble_deserialize
27 from tensorflow.contrib.boosted_trees.python.ops.gen_model_ops import tree_ensemble_serialize
29 from tensorflow.contrib.boosted_trees.python.ops.gen_model_ops import tree_ensemble_stamp_token
30 from tensorflow.contrib.boosted_trees.python.ops.gen_model_ops import tree_ensemble_used_handlers
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/models/
Dmultiple_additive_trees.cc21 namespace boosted_trees { namespace
25 const boosted_trees::trees::DecisionTreeEnsembleConfig& config, in Predict()
27 const boosted_trees::utils::BatchFeatures& features, in Predict()
51 const boosted_trees::trees::DecisionTreeConfig& tree = in Predict()
86 boosted_trees::utils::ParallelFor(batch_size, worker_threads->NumThreads(), in Predict()
/external/tensorflow/tensorflow/contrib/makefile/
Dtf_proto_files.txt1 tensorflow/contrib/boosted_trees/proto/learner.proto
2 tensorflow/contrib/boosted_trees/proto/quantiles.proto
3 tensorflow/contrib/boosted_trees/proto/split_info.proto
4 tensorflow/contrib/boosted_trees/proto/tree_config.proto
32 tensorflow/core/kernels/boosted_trees/boosted_trees.proto
/external/tensorflow/tensorflow/core/kernels/boosted_trees/quantiles/
Dweighted_quantiles_buffer_test.cc25 boosted_trees::quantiles::WeightedQuantilesBuffer<double, double>;
27 boosted_trees::quantiles::WeightedQuantilesBuffer<double,
35 boosted_trees::quantiles::WeightedQuantilesBuffer<double, double> in TEST_F()
41 boosted_trees::quantiles::WeightedQuantilesBuffer<double, double> in TEST_F()
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/quantiles/
Dweighted_quantiles_buffer_test.cc26 boosted_trees::quantiles::WeightedQuantilesBuffer<double, double>;
28 boosted_trees::quantiles::WeightedQuantilesBuffer<double,
36 boosted_trees::quantiles::WeightedQuantilesBuffer<double, double> in TEST_F()
42 boosted_trees::quantiles::WeightedQuantilesBuffer<double, double> in TEST_F()

1234567