DecisionTree.h File Reference

#include "../Matrix.h"
#include "../numerics/utils.h"
#include "Predictor.h"
#include <string>

Go to the source code of this file.

Classes
class	DecisionNode
struct	DataSplit
class	DecisionTree

Enumerations
enum	ImpurityMeasure { ENTROPY = 0 , GINI = 1 }
enum	DecisionNodeType { NONE = -1 , LEAF = 0 , DECISION = 1 }

Functions
const char *	measure_to_name (ImpurityMeasure measure)
double	entropy (const Matrix< double > &in)
Matrix< double >	count_bins (const Matrix< double > &in)
double	gini (const Matrix< double > &in)

Variables
struct DataSplit	DataSplit

Enumeration Type Documentation

DecisionNodeType

enum DecisionNodeType

Enumerator
NONE
LEAF
DECISION

ImpurityMeasure

enum ImpurityMeasure

Enumerator
ENTROPY
GINI

Function Documentation

count_bins()

Matrix< double > count_bins

(

const Matrix< double > &

in

)

entropy()

double entropy

(

const Matrix< double > &

in

)

gini()

double gini

(

const Matrix< double > &

in

)

measure_to_name()

const char * measure_to_name

(

ImpurityMeasure

measure

)

Variable Documentation

DataSplit

struct DataSplit DataSplit