Brush C++ API
A flexible interpretable machine learning framework
|
tree node specialization for Node. More...
#include <tree_node.h>
Public Member Functions | |
tree_node_ () | |
tree_node_ (const Node &val) | |
tree_node_ (Node &&val) | |
template<typename T > | |
auto | fit (const Dataset &d) |
template<typename T > | |
auto | predict (const Dataset &d, const float **weights=nullptr) |
template<typename T , typename W > | |
auto | predict (const Dataset &d, const W **weights) |
string | get_model (bool pretty=false) const |
string | get_tree_model (bool pretty=false, string offset="") const |
int | get_complexity () const |
int | get_size (bool include_weight=true) const |
Public Attributes | |
tree_node_< Node > * | parent |
tree_node_< Node > * | first_child |
tree_node_< Node > * | last_child |
tree_node_< Node > * | prev_sibling |
tree_node_< Node > * | next_sibling |
Node | data |
tree node specialization for Node.
Definition at line 22 of file tree_node.h.
|
inline |
Definition at line 24 of file tree_node.h.
|
inline |
Definition at line 28 of file tree_node.h.
|
inline |
Definition at line 32 of file tree_node.h.
int tree_node_< Node >::get_complexity | ( | ) | const |
string tree_node_< Node >::get_model | ( | bool | pretty = false | ) | const |
int tree_node_< Node >::get_size | ( | bool | include_weight = true | ) | const |
string tree_node_< Node >::get_tree_model | ( | bool | pretty = false, |
string | offset = "" ) const |
auto tree_node_< Node >::predict | ( | const Dataset & | d, |
const float ** | weights = nullptr ) |
Node tree_node_< Node >::data |
Definition at line 39 of file tree_node.h.
tree_node_<Node>* tree_node_< Node >::first_child |
Definition at line 37 of file tree_node.h.
tree_node_<Node> * tree_node_< Node >::last_child |
Definition at line 37 of file tree_node.h.
tree_node_<Node> * tree_node_< Node >::next_sibling |
Definition at line 38 of file tree_node.h.
tree_node_<Node>* tree_node_< Node >::parent |
Definition at line 36 of file tree_node.h.
tree_node_<Node>* tree_node_< Node >::prev_sibling |
Definition at line 38 of file tree_node.h.