Treelite
|
Public Types | |
using | PredFuncHandle = void * |
Public Member Functions | |
PredFunctionImpl (const SharedLibrary &library, int num_feature, int num_class) | |
TypeInfo | GetThresholdType () const override |
TypeInfo | GetLeafOutputType () const override |
size_t | PredictBatch (const DMatrix *dmat, size_t rbegin, size_t rend, bool pred_margin, PredictorOutputHandle out_pred) const override |
Additional Inherited Members | |
![]() | |
static std::unique_ptr< PredFunction > | Create (TypeInfo threshold_type, TypeInfo leaf_output_type, const SharedLibrary &library, int num_feature, int num_class) |
Definition at line 112 of file predictor.h.