Searched refs:class_weights_ (Results 1 – 2 of 2) sorted by relevance
635 auto it = class_weights_.find(c); in accumulate()636 if (it != class_weights_.end()) { in accumulate()639 smallest_weight_class_ = argmin(class_weights_); in accumulate()644 if (class_weights_.size() < n_) { in accumulate()645 class_weights_.insert(it, std::pair<int, float>(c, w)); in accumulate()646 if (class_weights_.size() == n_) { in accumulate()649 smallest_weight_class_ = argmin(class_weights_); in accumulate()658 it = class_weights_.find(smallest_weight_class_); in accumulate()660 class_weights_.erase(it); in accumulate()661 class_weights_[c] = new_weight; in accumulate()[all …]
455 std::unordered_map<int, float> class_weights_; variable