Method add

src/data/info_parser.cc:10–34

}

bool InfoParser::add(const Example& ex) {
  for (int i = 0; i < ex.slot_size(); ++i) {
    const auto& slot = ex.slot(i);
    if (slot.id() >= kSlotIDmax) return false;
    auto& sinfo = slot_info_[slot.id()];
    for (int j = 0; j < slot.key_size(); ++j) {
      uint64 key = slot.key(j);
      sinfo.set_min_key(std::min((uint64)sinfo.min_key(), key));
      sinfo.set_max_key(std::max((uint64)sinfo.max_key(), key + 1));
    }
    if (slot.key_size() > 0) {
      if (slot.val_size() == slot.key_size()) {
        sinfo.set_format(SlotInfo::SPARSE);
      } else {
        sinfo.set_format(SlotInfo::SPARSE_BINARY);
      }
    } else if (slot.val_size() > 0) {
      sinfo.set_format(SlotInfo::DENSE);
    }
    sinfo.set_nnz_ex(sinfo.nnz_ex() + 1);
    sinfo.set_nnz_ele(sinfo.nnz_ele() + std::max(slot.key_size(), slot.val_size()));
  }
  ++ num_ex_;
  return true;
}

ExampleInfo InfoParser::info() {
  info_.set_num_ex(num_ex_);