decisiontree.hpp File Reference

#include "algo_helper.h"
#include "flatnode.h"
#include <string>
#include <vector>

Include dependency graph for decisiontree.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	ML::DT::DecisionTreeParams
struct	ML::DT::TreeMetaDataNode< T, L >

Namespaces
	ML
	ML::DT

Typedefs
typedef TreeMetaDataNode< float, int >	ML::DT::TreeClassifierF
typedef TreeMetaDataNode< double, int >	ML::DT::TreeClassifierD
typedef TreeMetaDataNode< float, float >	ML::DT::TreeRegressorF
typedef TreeMetaDataNode< double, double >	ML::DT::TreeRegressorD

Functions
void	ML::DT::set_tree_params (DecisionTreeParams &params, int cfg_max_depth=-1, int cfg_max_leaves=-1, float cfg_max_features=1.0f, int cfg_max_n_bins=128, int cfg_min_samples_leaf=1, int cfg_min_samples_split=2, float cfg_min_impurity_decrease=0.0f, CRITERION cfg_split_criterion=CRITERION_END, int cfg_max_batch_size=4096)
	Set all DecisionTreeParams members. More...
template<class T , class L >
std::string	ML::DT::get_tree_summary_text (const TreeMetaDataNode< T, L > *tree)
	Obtain high-level tree information. More...
template<class T , class L >
std::string	ML::DT::get_tree_text (const TreeMetaDataNode< T, L > *tree)
	Obtain detailed tree information. More...
template<class T , class L >
std::string	ML::DT::get_tree_json (const TreeMetaDataNode< T, L > *tree)
	Export tree as a JSON string. More...