Treelite
Public Member Functions | Static Public Member Functions | List of all members
PredFunction Class Referenceabstract
Inheritance diagram for PredFunction:
Inheritance graph
[legend]

Public Member Functions

virtual TypeInfo GetThresholdType () const =0
 
virtual TypeInfo GetLeafOutputType () const =0
 
virtual size_t PredictBatch (const DMatrix *dmat, size_t rbegin, size_t rend, bool pred_margin, PredictorOutputHandle out_pred) const =0
 

Static Public Member Functions

static std::unique_ptr< PredFunctionCreate (TypeInfo threshold_type, TypeInfo leaf_output_type, const SharedLibrary &library, int num_feature, int num_class)
 

Detailed Description

Definition at line 48 of file predictor.h.


The documentation for this class was generated from the following files: