Home
last modified time | relevance | path

Searched refs:ClassificationStats (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dgrow_stats.cc68 ClassificationStats::ClassificationStats(const TensorForestParams& params, in ClassificationStats() function in tensorflow::tensorforest::ClassificationStats
132 void ClassificationStats::AdditionalInitializationExample( in AdditionalInitializationExample()
160 bool ClassificationStats::IsFinished() const { in IsFinished()
165 float ClassificationStats::MaybeCachedGiniScore(int split, float* left_sum, in MaybeCachedGiniScore()
182 void ClassificationStats::AddExample( in AddExample()
211 void ClassificationStats::CheckPrune() { in CheckPrune()
258 void ClassificationStats::CheckPruneHoeffding() { in CheckPruneHoeffding()
283 void ClassificationStats::CheckFinishEarly() { in CheckFinishEarly()
297 void ClassificationStats::CheckFinishEarlyHoeffding() { in CheckFinishEarlyHoeffding()
306 std::bind(&ClassificationStats::MaybeCachedGiniScore, this, in CheckFinishEarlyHoeffding()
[all …]
Dgrow_stats.h168 class ClassificationStats : public GrowStats {
170 ClassificationStats(const TensorForestParams& params, int32 depth);
299 class DenseClassificationGrowStats : public ClassificationStats {
302 : ClassificationStats(params, depth) {} in DenseClassificationGrowStats()
367 class SparseClassificationGrowStats : public ClassificationStats {
370 : ClassificationStats(params, depth) {} in SparseClassificationGrowStats()
459 class FixedSizeSparseClassificationGrowStats : public ClassificationStats {
463 : ClassificationStats(params, depth) {} in FixedSizeSparseClassificationGrowStats()