Searched refs:class_weights_ (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
H A Dgrow_stats.cc635 auto it = class_weights_.find(c);
636 if (it != class_weights_.end()) {
639 smallest_weight_class_ = argmin(class_weights_);
644 if (class_weights_.size() < n_) {
645 class_weights_.insert(it, std::pair<int, float>(c, w));
646 if (class_weights_.size() == n_) {
649 smallest_weight_class_ = argmin(class_weights_);
658 it = class_weights_.find(smallest_weight_class_);
660 class_weights_.erase(it);
661 class_weights_[
[all...]
H A Dgrow_stats.h452 // class_weights_.size() == n_.
455 std::unordered_map<int, float> class_weights_; member in class:tensorflow::tensorforest::FixedSizeClassStats

Completed in 256 milliseconds