Treelite
Public Types | Public Member Functions | List of all members
PredFunctionImpl< ThresholdType, LeafOutputType > Class Template Reference
Inheritance diagram for PredFunctionImpl< ThresholdType, LeafOutputType >:
Inheritance graph
[legend]
Collaboration diagram for PredFunctionImpl< ThresholdType, LeafOutputType >:
Collaboration graph
[legend]

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 Public Member Functions inherited from PredFunction
static std::unique_ptr< PredFunctionCreate (TypeInfo threshold_type, TypeInfo leaf_output_type, const SharedLibrary &library, int num_feature, int num_class)
 

Detailed Description

template<typename ThresholdType, typename LeafOutputType>
class treelite::predictor::PredFunctionImpl< ThresholdType, LeafOutputType >

Definition at line 75 of file predictor.h.


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