Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dgrow_stats_test.cc36 using tensorflow::tensorforest::SparseClassificationGrowStats;
326 std::unique_ptr<SparseClassificationGrowStats> stat( in TEST()
327 new SparseClassificationGrowStats(params, 1)); in TEST()
344 std::unique_ptr<SparseClassificationGrowStats> new_stat( in TEST()
345 new SparseClassificationGrowStats(params, 1)); in TEST()
Dgrow_stats.cc506 void SparseClassificationGrowStats::ExtractFromProto(const FertileSlot& slot) { in ExtractFromProto()
534 void SparseClassificationGrowStats::PackToProto(FertileSlot* slot) const { in PackToProto()
563 float SparseClassificationGrowStats::GiniScore(int split, float* left_sum, in GiniScore()
592 void SparseClassificationGrowStats::InitLeafClassStats( in InitLeafClassStats()
Dsplit_collection_operators.cc52 new SparseClassificationGrowStats(params_, depth)); in CreateGrowStats()
Dgrow_stats.h367 class SparseClassificationGrowStats : public ClassificationStats {
369 SparseClassificationGrowStats(const TensorForestParams& params, int32 depth) in SparseClassificationGrowStats() function