Function readMatrixInfo

src/data/common.cc:19–48

}

MatrixInfo readMatrixInfo(
    const ExampleInfo& info, int slot_id, int sizeof_idx, int sizeof_val) {
  MatrixInfo f;
  int i = 0;
  for (; i < info.slot_size(); ++i) if (info.slot(i).id() == slot_id) break;
  if (i == info.slot_size()) return f;

  auto slot = info.slot(i);
  if (slot.format() == SlotInfo::DENSE) {
    f.set_type(MatrixInfo::DENSE);
  } else if (slot.format() == SlotInfo::SPARSE) {
    f.set_type(MatrixInfo::SPARSE);
  } else if (slot.format() == SlotInfo::SPARSE_BINARY) {
    f.set_type(MatrixInfo::SPARSE_BINARY);
  }
  f.set_row_major(true);
  f.set_id(slot.id());
  f.mutable_row()->set_begin(0);
  f.mutable_row()->set_end(info.num_ex());
  f.mutable_col()->set_begin(slot.min_key());
  f.mutable_col()->set_end(slot.max_key());

  f.set_nnz(slot.nnz_ele());
  f.set_sizeof_index(sizeof_idx);
  f.set_sizeof_value(sizeof_val);

  // LL << info.DebugString() << "\n" << f.DebugString();
  // *f.mutable_ins_info() = info;
  return f;
}

ExampleInfo mergeExampleInfo(const ExampleInfo& A, const ExampleInfo& B) {
  std::map<int, SlotInfo> slots;