treelite
|
C API of treelite (runtime portion) More...
Go to the source code of this file.
Functions | |
int | TreeliteAssembleSparseBatch (const float *data, const uint32_t *col_ind, const size_t *row_ptr, size_t num_row, size_t num_col, CSRBatchHandle *out) |
assemble a sparse batch More... | |
int | TreeliteDeleteSparseBatch (CSRBatchHandle handle) |
delete a sparse batch from memory More... | |
int | TreeliteAssembleDenseBatch (const float *data, float missing_value, size_t num_row, size_t num_col, DenseBatchHandle *out) |
assemble a dense batch More... | |
int | TreeliteDeleteDenseBatch (DenseBatchHandle handle) |
delete a dense batch from memory More... | |
int | TreeliteBatchGetDimension (void *handle, int batch_sparse, size_t *out_num_row, size_t *out_num_col) |
get dimensions of a batch More... | |
int | TreelitePredictorLoad (const char *library_path, int num_worker_thread, int include_master_thread, PredictorHandle *out) |
load prediction code into memory. This function assumes that the prediction code has been already compiled into a dynamic shared library object (.so/.dll/.dylib). More... | |
int | TreelitePredictorPredictBatch (PredictorHandle handle, void *batch, int batch_sparse, int verbose, int pred_margin, float *out_result, size_t *out_result_size) |
Make predictions on a batch of data rows (synchronously). This function internally divides the workload among all worker threads. More... | |
int | TreelitePredictorQueryResultSize (PredictorHandle handle, void *batch, int batch_sparse, size_t *out) |
Given a batch of data rows, query the necessary size of array to hold predictions for all data points. More... | |
int | TreelitePredictorQueryNumOutputGroup (PredictorHandle handle, size_t *out) |
Get the number of output groups in the loaded model The number is 1 for most tasks; it is greater than 1 for multiclass classifcation. More... | |
int | TreelitePredictorFree (PredictorHandle handle) |
delete predictor from memory More... | |
C API of treelite (runtime portion)
Copyright (c) 2017 by Contributors
Definition in file c_api_runtime.cc.