#include <dataset.h>
Collaboration diagram for ML::DT::Dataset< DataT, LabelT, IdxT >:
Public Attributes | |
| const DataT * | data |
| const LabelT * | labels |
| IdxT | M |
| IdxT | N |
| IdxT | n_sampled_rows |
| IdxT | n_sampled_cols |
| IdxT * | row_ids |
| IdxT | num_outputs |
Member Data Documentation
◆ data
template<typename DataT , typename LabelT , typename IdxT >
| const DataT* ML::DT::Dataset< DataT, LabelT, IdxT >::data |
input dataset (assumed to be col-major)
◆ labels
template<typename DataT , typename LabelT , typename IdxT >
| const LabelT* ML::DT::Dataset< DataT, LabelT, IdxT >::labels |
input labels
◆ M
template<typename DataT , typename LabelT , typename IdxT >
| IdxT ML::DT::Dataset< DataT, LabelT, IdxT >::M |
total rows in dataset
◆ N
template<typename DataT , typename LabelT , typename IdxT >
| IdxT ML::DT::Dataset< DataT, LabelT, IdxT >::N |
total cols in dataset
◆ n_sampled_cols
template<typename DataT , typename LabelT , typename IdxT >
| IdxT ML::DT::Dataset< DataT, LabelT, IdxT >::n_sampled_cols |
total sampled cols in dataset
◆ n_sampled_rows
template<typename DataT , typename LabelT , typename IdxT >
| IdxT ML::DT::Dataset< DataT, LabelT, IdxT >::n_sampled_rows |
total sampled rows in dataset
◆ num_outputs
template<typename DataT , typename LabelT , typename IdxT >
| IdxT ML::DT::Dataset< DataT, LabelT, IdxT >::num_outputs |
Number of classes or regression outputs
◆ row_ids
template<typename DataT , typename LabelT , typename IdxT >
| IdxT* ML::DT::Dataset< DataT, LabelT, IdxT >::row_ids |
indices of sampled rows
The documentation for this struct was generated from the following file:
- src/decisiontree/batched-levelalgo/dataset.h
