Treelite
Functions
predict.cc File Reference

General Tree Inference Library (GTIL), providing a reference implementation for predicting with decision trees. GTIL is useful in cases it is infeasible to build the tree models as native shared libs. More...

#include <treelite/gtil.h>
#include <treelite/tree.h>
#include <treelite/data.h>
#include <dmlc/logging.h>
#include <limits>
#include <vector>
#include <cstddef>
#include "./pred_transform.h"
Include dependency graph for predict.cc:

Go to the source code of this file.

Functions

std::size_t Predict (const Model *model, const DMatrix *input, float *output, bool pred_transform=true)
 
std::size_t Predict (const Model *model, const float *input, std::size_t num_row, float *output, bool pred_transform=true)
 
std::size_t GetPredictOutputSize (const Model *model, std::size_t num_row)
 
std::size_t GetPredictOutputSize (const Model *model, const DMatrix *input)
 

Detailed Description

General Tree Inference Library (GTIL), providing a reference implementation for predicting with decision trees. GTIL is useful in cases it is infeasible to build the tree models as native shared libs.

Copyright (c) 2021 by Contributors

Author
Hyunsu Cho

Definition in file predict.cc.