dataset.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2019-2022, NVIDIA CORPORATION.
 * SPDX-License-Identifier: Apache-2.0
 */
 
#pragma once
 
namespace ML {
namespace DT {
 
template <typename DataT, typename LabelT, typename IdxT>
struct Dataset {
  const DataT* data;
  const LabelT* labels;
  IdxT M;
  IdxT N;
  IdxT n_sampled_rows;
  IdxT n_sampled_cols;
  IdxT* row_ids;
  IdxT num_outputs;
};
 
}  // namespace DT
}  // namespace ML