Public Member Functions | |
std::vector< int > | decide (const std::vector< std::vector< float > > &data) |
float | split (const std::vector< std::vector< float > > &data, const std::vector< int > &labels, const std::vector< float > &weights, rutz::shared_ptr< DecisionNode > &left, rutz::shared_ptr< DecisionNode > &right, const rutz::shared_ptr< DecisionNode > parent=rutz::shared_ptr< DecisionNode >(NULL)) |
size_t | getDim () |
bool | isValid () |
int | printNode (std::string &output, int depth=0) |
void | writeNode (std::ostream &outstream, bool needEnd=true) |
rutz::shared_ptr< DecisionNode > | readNode (std::istream &instream) |
void | setDim (size_t dim) |
void | setLeaf (bool isLeaf) |
void | setParent (rutz::shared_ptr< DecisionNode > parent) |
void | setLeftConstraint (float constraint) |
void | setRightConstraint (float constraint) |
void | setClass (int classId) |
int | getClass () |
Definition at line 68 of file DecisionTree.H.