Treelite
Functions
Dmatrix

Functions

int TreeliteDMatrixCreateFromCSR (const void *data, const char *data_type, const uint32_t *col_ind, const size_t *row_ptr, size_t num_row, size_t num_col, DMatrixHandle *out)
 create DMatrix from a (in-memory) CSR matrix More...
 
int TreeliteDMatrixCreateFromMat (const void *data, const char *data_type, size_t num_row, size_t num_col, const void *missing_value, DMatrixHandle *out)
 create DMatrix from a (in-memory) dense matrix More...
 
int TreeliteDMatrixGetDimension (DMatrixHandle handle, size_t *out_num_row, size_t *out_num_col, size_t *out_nelem)
 get dimensions of a DMatrix More...
 
int TreeliteDMatrixFree (DMatrixHandle handle)
 delete DMatrix from memory More...
 

Detailed Description

Data matrix interface

Function Documentation

◆ TreeliteDMatrixCreateFromCSR()

int TreeliteDMatrixCreateFromCSR ( const void *  data,
const char *  data_type,
const uint32_t *  col_ind,
const size_t *  row_ptr,
size_t  num_row,
size_t  num_col,
DMatrixHandle out 
)

create DMatrix from a (in-memory) CSR matrix

Parameters
datafeature values
data_typeType of data elements
col_indfeature indices
row_ptrpointer to row headers
num_rownumber of rows
num_colnumber of columns
outthe created DMatrix
Returns
0 for success, -1 for failure

Definition at line 32 of file c_api_common.cc.

◆ TreeliteDMatrixCreateFromMat()

int TreeliteDMatrixCreateFromMat ( const void *  data,
const char *  data_type,
size_t  num_row,
size_t  num_col,
const void *  missing_value,
DMatrixHandle out 
)

create DMatrix from a (in-memory) dense matrix

Parameters
datafeature values
data_typeType of data elements
num_rownumber of rows
num_colnumber of columns
missing_valuevalue to represent missing value
outthe created DMatrix
Returns
0 for success, -1 for failure

Definition at line 43 of file c_api_common.cc.

◆ TreeliteDMatrixFree()

int TreeliteDMatrixFree ( DMatrixHandle  handle)

delete DMatrix from memory

Parameters
handlehandle to DMatrix
Returns
0 for success, -1 for failure

Definition at line 66 of file c_api_common.cc.

◆ TreeliteDMatrixGetDimension()

int TreeliteDMatrixGetDimension ( DMatrixHandle  handle,
size_t *  out_num_row,
size_t *  out_num_col,
size_t *  out_nelem 
)

get dimensions of a DMatrix

Parameters
handlehandle to DMatrix
out_num_rowused to set number of rows
out_num_colused to set number of columns
out_nelemused to set number of nonzero entries
Returns
0 for success, -1 for failure

Definition at line 54 of file c_api_common.cc.