Searched refs:GrowStats (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| D | split_collection_operators.cc | 43 std::unique_ptr<GrowStats> SplitCollectionOperator::CreateGrowStats( in CreateGrowStats()
47 return std::unique_ptr<GrowStats>( in CreateGrowStats()
51 return std::unique_ptr<GrowStats>( in CreateGrowStats()
55 return std::unique_ptr<GrowStats>( in CreateGrowStats()
59 return std::unique_ptr<GrowStats>( in CreateGrowStats()
89 stats_[node_id] = std::unique_ptr<GrowStats>(CreateGrowStats(node_id, depth)); in InitializeSlot()
|
| D | grow_stats.h | 35 class GrowStats {
37 virtual ~GrowStats() {} in ~GrowStats()
87 GrowStats(const TensorForestParams& params, int32 depth);
115 class SimpleStats : public GrowStats {
118 : GrowStats(params, depth) {} in SimpleStats()
168 class ClassificationStats : public GrowStats {
527 class LeastSquaresRegressionGrowStats : public GrowStats {
530 : GrowStats(params, depth) {} in LeastSquaresRegressionGrowStats()
|
| D | split_collection_operators.h | 41 virtual std::unique_ptr<GrowStats> CreateGrowStats(int32 node_id,
91 std::unordered_map<int32, std::unique_ptr<GrowStats>> stats_;
|
| D | graph_collection_operator.cc | 27 std::unique_ptr<GrowStats> GraphRunnerSplitCollectionOperator::CreateGrowStats( in CreateGrowStats()
29 return std::unique_ptr<GrowStats>(new SimpleStats(params_, depth)); in CreateGrowStats()
|
| D | graph_collection_operator.h | 48 std::unique_ptr<GrowStats> CreateGrowStats(int32 node_id,
|
| D | grow_stats.cc | 32 GrowStats::GrowStats(const TensorForestParams& params, int32 depth) in GrowStats() function in tensorflow::tensorforest::GrowStats
41 void GrowStats::AddSplit(const decision_trees::BinaryNode& split, in AddSplit()
60 void GrowStats::RemoveSplit(int split_num) { in RemoveSplit()
70 : GrowStats(params, depth), finish_early_(false) { in ClassificationStats()
|
| D | grow_stats_test.cc | 34 using tensorflow::tensorforest::GrowStats;
55 void RunBatch(GrowStats* stats, const TestableInputTarget* target) { in RunBatch()
|