#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.