Treelite
Public Member Functions | Public Attributes | Friends | List of all members
CSRDMatrixImpl< ElementType > Class Template Reference
Inheritance diagram for CSRDMatrixImpl< ElementType >:
Inheritance graph
[legend]
Collaboration diagram for CSRDMatrixImpl< ElementType >:
Collaboration graph
[legend]

Public Member Functions

 CSRDMatrixImpl (std::vector< ElementType > data, std::vector< uint32_t > col_ind, std::vector< size_t > row_ptr, size_t num_row, size_t num_col)
 
 CSRDMatrixImpl (const CSRDMatrixImpl &)=default
 
 CSRDMatrixImpl (CSRDMatrixImpl &&) noexcept=default
 
CSRDMatrixImploperator= (const CSRDMatrixImpl &)=default
 
CSRDMatrixImploperator= (CSRDMatrixImpl &&) noexcept=default
 
size_t GetNumRow () const override
 
size_t GetNumCol () const override
 
size_t GetNumElem () const override
 
DMatrixType GetType () const override
 
template<typename OutputType >
void FillRow (size_t row_id, OutputType *out) const
 
template<typename OutputType >
void ClearRow (size_t row_id, OutputType *out) const
 
- Public Member Functions inherited from CSRDMatrix
TypeInfo GetElementType () const override
 

Public Attributes

std::vector< ElementType > data
 feature values
 
std::vector< uint32_t > col_ind
 feature indices. col_ind[i] indicates the feature index associated with data[i].
 
std::vector< size_t > row_ptr
 pointer to row headers; length is [num_row] + 1.
 
size_t num_row
 number of rows
 
size_t num_col
 number of columns (i.e. # of features used)
 

Friends

class CSRDMatrix
 

Additional Inherited Members

- Static Public Member Functions inherited from CSRDMatrix
template<typename ElementType >
static std::unique_ptr< CSRDMatrixCreate (std::vector< ElementType > data, std::vector< uint32_t > col_ind, std::vector< size_t > row_ptr, size_t num_row, size_t num_col)
 
template<typename ElementType >
static std::unique_ptr< CSRDMatrixCreate (const void *data, const uint32_t *col_ind, const size_t *row_ptr, size_t num_row, size_t num_col)
 
static std::unique_ptr< CSRDMatrixCreate (TypeInfo type, const void *data, const uint32_t *col_ind, const size_t *row_ptr, size_t num_row, size_t num_col)
 
static std::unique_ptr< CSRDMatrixCreate (const char *filename, const char *format, const char *data_type, int nthread, int verbose)
 

Detailed Description

template<typename ElementType>
class treelite::CSRDMatrixImpl< ElementType >

Definition at line 112 of file data.h.


The documentation for this class was generated from the following files: