Public Member Functions | |
DecisionTree (int maxSplits=1) | |
void | train (const std::vector< std::vector< float > > &data, const std::vector< int > &labels, std::vector< float > weights=std::vector< float >()) |
std::vector< int > | predict (const std::vector< std::vector< float > > &data, std::vector< float > weights=std::vector< float >()) |
void | printTree () |
std::deque< rutz::shared_ptr < DecisionNode > > | getNodes () |
void | addNode (rutz::shared_ptr< DecisionNode > node) |
Definition at line 51 of file DecisionTree.H.