OmniSciDB  a5dc49c757
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
AbstractTreeModel Class Referenceabstract

#include <MLModel.h>

+ Inheritance diagram for AbstractTreeModel:
+ Collaboration diagram for AbstractTreeModel:

Public Member Functions

virtual MLModelType getModelType () const =0
 
virtual std::string getModelTypeString () const =0
 
virtual int64_t getNumFeatures () const =0
 
virtual int64_t getNumTrees () const =0
 
virtual ~AbstractTreeModel ()=default
 
virtual void traverseDF (const int64_t tree_idx, TreeModelVisitor &tree_node_visitor) const =0
 
- Public Member Functions inherited from AbstractMLModel
 AbstractMLModel (const std::string &model_metadata)
 
 AbstractMLModel (const std::string &model_metadata, const std::vector< std::vector< std::string >> &cat_feature_keys)
 
virtual ~AbstractMLModel ()=default
 
const std::string & getModelMetadataStr () const
 
MLModelMetadata getModelMetadata () const
 
const std::vector< std::vector
< std::string > > & 
getCatFeatureKeys () const
 
const int64_t getNumCatFeatures () const
 
const int64_t getNumOneHotFeatures () const
 
const int64_t getNumLogicalFeatures () const
 

Additional Inherited Members

- Protected Attributes inherited from AbstractMLModel
std::string model_metadata_
 
std::vector< std::vector
< std::string > > 
cat_feature_keys_
 

Detailed Description

Definition at line 156 of file MLModel.h.

Constructor & Destructor Documentation

virtual AbstractTreeModel::~AbstractTreeModel ( )
virtualdefault

Member Function Documentation

virtual MLModelType AbstractTreeModel::getModelType ( ) const
pure virtual

Implements AbstractMLModel.

virtual std::string AbstractTreeModel::getModelTypeString ( ) const
pure virtual

Implements AbstractMLModel.

virtual int64_t AbstractTreeModel::getNumFeatures ( ) const
pure virtual

Implements AbstractMLModel.

virtual int64_t AbstractTreeModel::getNumTrees ( ) const
pure virtual
virtual void AbstractTreeModel::traverseDF ( const int64_t  tree_idx,
TreeModelVisitor &  tree_node_visitor 
) const
pure virtual

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