12 #ifndef TREELITE_C_API_RUNTIME_H_ 13 #define TREELITE_C_API_RUNTIME_H_ 171 TREELITE_DLL
int TreelitePredictorQueryThresholdType(
PredictorHandle handle,
const char** out);
172 TREELITE_DLL
int TreelitePredictorQueryLeafOutputType(
PredictorHandle handle,
const char** out);
181 #endif // TREELITE_C_API_RUNTIME_H_ int TreeliteCreatePredictorOutputVector(PredictorHandle handle, DMatrixHandle batch, PredictorOutputHandle *out_output_vector)
Convenience function to allocate an output vector that is able to hold the prediction result for a gi...
int TreelitePredictorPredictBatch(PredictorHandle handle, DMatrixHandle batch, int verbose, int pred_margin, PredictorOutputHandle out_result, size_t *out_result_size)
Make predictions on a batch of data rows (synchronously). This function internally divides the worklo...
void * PredictorOutputHandle
handle to output from predictor
int TreelitePredictorQueryPredTransform(PredictorHandle handle, const char **out)
Get name of post prediction transformation used to train the loaded model.
int TreelitePredictorQueryNumFeature(PredictorHandle handle, size_t *out)
Get the width (number of features) of each instance used to train the loaded model.
void * DMatrixHandle
handle to a data matrix
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 com...
int TreelitePredictorQueryRatioC(PredictorHandle handle, float *out)
Get c value of exponential standard ratio transformation used to train the loaded model...
C API of Treelite, used for interfacing with other languages This header is used by both the runtime ...
void * PredictorHandle
handle to predictor class
int TreeliteDeletePredictorOutputVector(PredictorHandle handle, PredictorOutputHandle output_vector)
De-allocate an output vector.
int TreelitePredictorFree(PredictorHandle handle)
delete predictor from memory
int TreelitePredictorQueryNumClass(PredictorHandle handle, size_t *out)
Get the number classes in the loaded model The number is 1 for most tasks; it is greater than 1 for m...
int TreelitePredictorQueryResultSize(PredictorHandle handle, DMatrixHandle batch, size_t *out)
Given a batch of data rows, query the necessary size of array to hold predictions for all data points...
int TreelitePredictorQuerySigmoidAlpha(PredictorHandle handle, float *out)
Get alpha value of sigmoid transformation used to train the loaded model.
int TreelitePredictorQueryGlobalBias(PredictorHandle handle, float *out)
Get global bias which adjusting predicted margin scores.