Treelite
Functions
c_api_runtime.cc File Reference

C API of treelite (runtime portion) More...

#include <treelite/predictor.h>
#include <treelite/c_api_runtime.h>
#include <dmlc/thread_local.h>
#include <string>
#include <cstring>
#include "./c_api_error.h"
Include dependency graph for c_api_runtime.cc:

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, 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 TreelitePredictorPredictInst (PredictorHandle handle, union TreelitePredictorEntry *inst, int pred_margin, float *out_result, size_t *out_result_size)
 Make predictions on a single data row (synchronously). The work will be scheduled to the calling thread. 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 TreelitePredictorQueryResultSizeSingleInst (PredictorHandle handle, size_t *out)
 Query the necessary size of array to hold the prediction for a single data row. 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 TreelitePredictorQueryNumFeature (PredictorHandle handle, size_t *out)
 Get the width (number of features) of each instance used to train the loaded model. More...
 
int TreelitePredictorQueryPredTransform (PredictorHandle handle, const char **out)
 Get name of post prediction transformation used to train the loaded model. More...
 
int TreelitePredictorQuerySigmoidAlpha (PredictorHandle handle, float *out)
 Get alpha value of sigmoid transformation used to train the loaded model. More...
 
int TreelitePredictorQueryGlobalBias (PredictorHandle handle, float *out)
 Get global bias which adjusting predicted margin scores. More...
 
int TreelitePredictorFree (PredictorHandle handle)
 delete predictor from memory More...
 

Detailed Description

C API of treelite (runtime portion)

Copyright (c) 2017-2020 by Contributors

Author
Hyunsu Cho

Definition in file c_api_runtime.cc.