Home
last modified time | relevance | path

Searched refs:DenseClassificationGrowStats (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dgrow_stats_test.cc30 using tensorflow::tensorforest::DenseClassificationGrowStats;
73 std::unique_ptr<DenseClassificationGrowStats> stat( in TEST()
74 new DenseClassificationGrowStats(params, 1)); in TEST()
90 std::unique_ptr<DenseClassificationGrowStats> new_stat( in TEST()
91 new DenseClassificationGrowStats(params, 1)); in TEST()
99 class TestableRunningStats : public DenseClassificationGrowStats {
102 : DenseClassificationGrowStats(params, depth) {} in TestableRunningStats()
143 std::unique_ptr<DenseClassificationGrowStats> new_stat( in TEST()
144 new DenseClassificationGrowStats(params, 1)); in TEST()
152 class TestableFinishEarly : public DenseClassificationGrowStats {
[all …]
Dgrow_stats.cc415 void DenseClassificationGrowStats::ExtractFromProto(const FertileSlot& slot) { in ExtractFromProto()
445 void DenseClassificationGrowStats::PackToProto(FertileSlot* slot) const { in PackToProto()
468 float DenseClassificationGrowStats::GiniScore(int split, float* left_sum, in GiniScore()
489 void DenseClassificationGrowStats::InitLeafClassStats( in InitLeafClassStats()
Dsplit_collection_operators.cc48 new DenseClassificationGrowStats(params_, depth)); in CreateGrowStats()
Dgrow_stats.h299 class DenseClassificationGrowStats : public ClassificationStats {
301 DenseClassificationGrowStats(const TensorForestParams& params, int32 depth) in DenseClassificationGrowStats() function