Treelite
|
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... | |
Data matrix interface
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
data | feature values |
data_type | Type of data elements |
col_ind | feature indices |
row_ptr | pointer to row headers |
num_row | number of rows |
num_col | number of columns |
out | the created DMatrix |
Definition at line 32 of file c_api_common.cc.
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
data | feature values |
data_type | Type of data elements |
num_row | number of rows |
num_col | number of columns |
missing_value | value to represent missing value |
out | the created DMatrix |
Definition at line 43 of file c_api_common.cc.
int TreeliteDMatrixFree | ( | DMatrixHandle | handle | ) |
delete DMatrix from memory
handle | handle to DMatrix |
Definition at line 66 of file c_api_common.cc.
int TreeliteDMatrixGetDimension | ( | DMatrixHandle | handle, |
size_t * | out_num_row, | ||
size_t * | out_num_col, | ||
size_t * | out_nelem | ||
) |
get dimensions of a DMatrix
handle | handle to DMatrix |
out_num_row | used to set number of rows |
out_num_col | used to set number of columns |
out_nelem | used to set number of nonzero entries |
Definition at line 54 of file c_api_common.cc.