Searched refs:boosted_trees (Results 1 - 25 of 97) sorted by relevance

1234

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/
H A Drandom_tree_gen.h20 #include "tensorflow/contrib/boosted_trees/proto/tree_config.pb.h" // NOLINT
25 namespace boosted_trees { namespace in namespace:tensorflow
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
[all...]
H A Dbatch_features_testutil.h18 #include "tensorflow/contrib/boosted_trees/lib/utils/batch_features.h"
23 namespace boosted_trees { namespace in namespace:tensorflow
39 boosted_trees::utils::BatchFeatures* batch_features);
42 } // namespace boosted_trees
/external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/
H A Dsplit_handler_ops.py21 from tensorflow.contrib.boosted_trees.python.ops import boosted_trees_ops_loader
24 from tensorflow.contrib.boosted_trees.python.ops.gen_split_handler_ops import *
H A Dtraining_ops.py21 from tensorflow.contrib.boosted_trees.python.ops import boosted_trees_ops_loader
24 from tensorflow.contrib.boosted_trees.python.ops.gen_training_ops import *
H A 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_examples
23 from tensorflow.contrib.boosted_trees.python.ops.gen_prediction_ops import gradient_trees_prediction
H A Dmodel_ops.py21 from tensorflow.contrib.boosted_trees.python.ops import boosted_trees_ops_loader
23 from tensorflow.contrib.boosted_trees.python.ops import gen_model_ops
24 from tensorflow.contrib.boosted_trees.python.ops.gen_model_ops import tree_ensemble_deserialize
25 from tensorflow.contrib.boosted_trees.python.ops.gen_model_ops import tree_ensemble_serialize
27 from tensorflow.contrib.boosted_trees.python.ops.gen_model_ops import tree_ensemble_stamp_token
/external/tensorflow/tensorflow/contrib/boosted_trees/
H A D__init__.py21 from tensorflow.contrib.boosted_trees.python import *
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
H A D__init__.py21 from tensorflow.contrib.boosted_trees.estimator_batch import *
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/models/
H A Dmultiple_additive_trees.h20 #include "tensorflow/contrib/boosted_trees/lib/utils/batch_features.h"
21 #include "tensorflow/contrib/boosted_trees/proto/tree_config.pb.h" // NOLINT
27 namespace boosted_trees { namespace in namespace:tensorflow
37 const boosted_trees::trees::DecisionTreeEnsembleConfig& config,
39 const boosted_trees::utils::BatchFeatures& features,
45 } // namespace boosted_trees
H A Dmultiple_additive_trees.cc15 #include "tensorflow/contrib/boosted_trees/lib/models/multiple_additive_trees.h"
16 #include "tensorflow/contrib/boosted_trees/lib/trees/decision_tree.h"
17 #include "tensorflow/contrib/boosted_trees/lib/utils/batch_features.h"
18 #include "tensorflow/contrib/boosted_trees/lib/utils/parallel_for.h"
21 namespace boosted_trees { namespace in namespace:tensorflow
25 const boosted_trees::trees::DecisionTreeEnsembleConfig& config,
27 const boosted_trees::utils::BatchFeatures& features,
45 const boosted_trees::trees::DecisionTreeConfig& tree =
76 boosted_trees::utils::ParallelFor(batch_size, worker_threads->NumThreads(),
81 } // namespace boosted_trees
[all...]
/external/tensorflow/tensorflow/contrib/boosted_trees/python/
H A D__init__.py21 from tensorflow.contrib.boosted_trees.python.ops import *
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/partitioners/
H A Dexample_partitioner.cc15 #include "tensorflow/contrib/boosted_trees/lib/learner/common/partitioners/example_partitioner.h"
16 #include "tensorflow/contrib/boosted_trees/lib/utils/parallel_for.h"
19 namespace boosted_trees { namespace in namespace:tensorflow
23 const boosted_trees::trees::DecisionTreeConfig& tree_config,
24 const boosted_trees::utils::BatchFeatures& features,
41 example_partition = boosted_trees::trees::DecisionTree::Traverse(
51 boosted_trees::utils::ParallelFor(batch_size, desired_parallelism,
56 const boosted_trees::trees::DecisionTreeConfig& tree_config,
57 const boosted_trees::utils::BatchFeatures& features,
73 uint32 partition = boosted_trees
[all...]
H A Dexample_partitioner_test.cc15 #include "tensorflow/contrib/boosted_trees/lib/learner/common/partitioners/example_partitioner.h"
21 namespace boosted_trees { namespace in namespace:tensorflow
37 boosted_trees::utils::BatchFeatures batch_features_;
41 boosted_trees::trees::DecisionTreeConfig tree_config;
53 boosted_trees::trees::DecisionTreeConfig tree_config;
76 boosted_trees::trees::DecisionTreeConfig tree_config;
98 } // namespace boosted_trees
H A Dexample_partitioner.h19 #include "tensorflow/contrib/boosted_trees/lib/trees/decision_tree.h"
20 #include "tensorflow/contrib/boosted_trees/lib/utils/batch_features.h"
24 namespace boosted_trees { namespace in namespace:tensorflow
50 } // namespace boosted_trees
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/stats/
H A Dfeature-split-candidate.h19 #include "tensorflow/contrib/boosted_trees/lib/learner/common/stats/split-stats.h"
20 #include "tensorflow/contrib/boosted_trees/proto/tree_config.pb.h"
23 namespace boosted_trees { namespace in namespace:tensorflow
36 const boosted_trees::trees::TreeNode& node,
47 boosted_trees::trees::TreeNode tree_node;
58 } // namespace boosted_trees
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/quantiles/
H A Dweighted_quantiles_buffer_test.cc15 #include "tensorflow/contrib/boosted_trees/lib/quantiles/weighted_quantiles_buffer.h"
26 boosted_trees::quantiles::WeightedQuantilesBuffer<double, double>;
28 boosted_trees::quantiles::WeightedQuantilesBuffer<double,
36 boosted_trees::quantiles::WeightedQuantilesBuffer<double, double>
42 boosted_trees::quantiles::WeightedQuantilesBuffer<double, double>
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/trees/
H A Ddecision_tree.h18 #include "tensorflow/contrib/boosted_trees/lib/utils/example.h"
19 #include "tensorflow/contrib/boosted_trees/proto/tree_config.pb.h" // NOLINT
22 namespace boosted_trees { namespace in namespace:tensorflow
46 } // namespace boosted_trees
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
H A Dparallel_for.h21 namespace boosted_trees { namespace in namespace:tensorflow
30 } // namespace boosted_trees
H A Drandom.h21 namespace boosted_trees { namespace in namespace:tensorflow
36 } // namespace boosted_trees
H A Doptional_value.h22 namespace boosted_trees { namespace in namespace:tensorflow
44 } // namespace boosted_trees
H A Ddropout_utils.h21 #include "tensorflow/contrib/boosted_trees/proto/learner.pb.h" // NOLINT
26 namespace boosted_trees { namespace in namespace:tensorflow
74 } // namespace boosted_trees
/external/tensorflow/tensorflow/contrib/boosted_trees/resources/
H A Dstamped_resource.h22 namespace boosted_trees { namespace in namespace:tensorflow
40 } // namespace boosted_trees
H A Ddecision_tree_ensemble_resource.h18 #include "tensorflow/contrib/boosted_trees/lib/trees/decision_tree.h"
19 #include "tensorflow/contrib/boosted_trees/resources/stamped_resource.h"
25 namespace boosted_trees { namespace in namespace:tensorflow
35 boosted_trees::trees::DecisionTreeEnsembleConfig>(&arena_)) {}
42 const boosted_trees::trees::DecisionTreeEnsembleConfig&
65 boosted_trees::trees::GrowingMetadata* const growing_metadata =
76 boosted_trees::trees::DecisionTreeConfig* AddNewTree(const float weight) {
79 boosted_trees::trees::DecisionTreeMetadata* const metadata =
92 boosted_trees::trees::DecisionTreeConfig* LastTree() {
98 boosted_trees
[all...]
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
H A Dmodel_ops.cc17 #include "tensorflow/contrib/boosted_trees/lib/utils/tensor_utils.h"
18 #include "tensorflow/contrib/boosted_trees/resources/decision_tree_ensemble_resource.h"
27 namespace boosted_trees { namespace in namespace:tensorflow
29 using boosted_trees::models::DecisionTreeEnsembleResource;
47 auto* result = new boosted_trees::models::DecisionTreeEnsembleResource();
72 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource;
91 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource;
115 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource;
144 IsResourceInitialized<boosted_trees::models::DecisionTreeEnsembleResource>);
158 } // namespace boosted_trees
[all...]
H A Dtraining_ops.cc15 #include "tensorflow/contrib/boosted_trees/lib/utils/dropout_utils.h"
16 #include "tensorflow/contrib/boosted_trees/proto/learner.pb.h"
17 #include "tensorflow/contrib/boosted_trees/proto/split_info.pb.h"
18 #include "tensorflow/contrib/boosted_trees/resources/decision_tree_ensemble_resource.h"
23 using tensorflow::boosted_trees::learner::LearningRateDropoutDrivenConfig;
25 namespace boosted_trees { namespace in namespace:tensorflow
29 using boosted_trees::learner::LearningRateConfig;
30 using boosted_trees::trees::Leaf;
31 using boosted_trees::trees::TreeNode;
32 using boosted_trees
[all...]

Completed in 287 milliseconds

1234