#include <random_forest.h>
Public Member Functions | |
int | classifyInstance (SensorPoint::Ptr inst) |
void | growTree (vector< SensorPoint::Ptr > *c_dataset, vector< int > *inds) |
RandomTree (int c_d_tree_num) | |
RandomTree (RandomTreeMsg::Ptr) | |
void | writeTree (string &bag_file, bool is_first) |
Public Attributes | |
int | d_tree_num |
vector< SensorPoint::Ptr > * | dataset |
bool | is_abs |
int | num_classes |
Protected Member Functions | |
bool | attrCompare (int inst_i, int inst_j, int attr) |
void | findBestSplit (vector< int > *insts, vector< int > &attrs, pair< int, float > &ret) |
void | splitNode (vector< int > *node_inds, pair< int, float > &split_pt, pair< vector< int > *, vector< int > * > &split_nodes) |
Protected Attributes | |
RandomTreeMsg::Ptr | rand_tree |
Definition at line 51 of file random_forest.h.
collision_detection::RandomTree::RandomTree | ( | int | c_d_tree_num | ) |
Definition at line 15 of file random_forest.cpp.
collision_detection::RandomTree::RandomTree | ( | RandomTreeMsg::Ptr | r_tree | ) |
Definition at line 19 of file random_forest.cpp.
bool collision_detection::RandomTree::attrCompare | ( | int | inst_i, |
int | inst_j, | ||
int | attr | ||
) | [protected] |
Definition at line 24 of file random_forest.cpp.
int collision_detection::RandomTree::classifyInstance | ( | SensorPoint::Ptr | inst | ) |
Definition at line 347 of file random_forest.cpp.
void collision_detection::RandomTree::findBestSplit | ( | vector< int > * | insts, |
vector< int > & | attrs, | ||
pair< int, float > & | ret | ||
) | [protected] |
Definition at line 41 of file random_forest.cpp.
void collision_detection::RandomTree::growTree | ( | vector< SensorPoint::Ptr > * | c_dataset, |
vector< int > * | inds | ||
) |
Definition at line 130 of file random_forest.cpp.
void collision_detection::RandomTree::splitNode | ( | vector< int > * | node_inds, |
pair< int, float > & | split_pt, | ||
pair< vector< int > *, vector< int > * > & | split_nodes | ||
) | [protected] |
Definition at line 113 of file random_forest.cpp.
void collision_detection::RandomTree::writeTree | ( | string & | bag_file, |
bool | is_first | ||
) |
Definition at line 334 of file random_forest.cpp.
Definition at line 53 of file random_forest.h.
vector< SensorPoint::Ptr >* collision_detection::RandomTree::dataset |
Definition at line 55 of file random_forest.h.
Definition at line 64 of file random_forest.h.
Definition at line 54 of file random_forest.h.
RandomTreeMsg::Ptr collision_detection::RandomTree::rand_tree [protected] |
Definition at line 68 of file random_forest.h.