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

Public Member Functions

 QuantizerNode (const std::vector< std::vector< ThresholdType >> &cut_pts)
 
 QuantizerNode (std::vector< std::vector< ThresholdType >> &&cut_pts)
 
std::string GetDump () const override
 

Public Attributes

std::vector< std::vector< ThresholdType > > cut_pts
 
- Public Attributes inherited from ASTNode
ASTNodeparent
 
std::vector< ASTNode * > children
 
int node_id
 
int tree_id
 
dmlc::optional< size_t > data_count
 
dmlc::optional< double > sum_hess
 

Detailed Description

template<typename ThresholdType>
class treelite::compiler::QuantizerNode< ThresholdType >

Definition at line 64 of file ast.h.


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