DecisionTreeClassifier class
#include <DecisionTreeClassifier.h>

Contents

Base classes

class DecisionTreeBase
DecisionTreeBase is a semi-virtual parent class that contains methods used in both DecisionTreeClassifier and DecisionTreeRegressor.

Constructors, destructors, conversion operators

DecisionTreeClassifier(std::string lossCriterion, double maxFeatures = 1.0, int minSamplesSplit = 2, int maxDepth = -1, int minSamplesLeaf = 0, double minImpurityDecrease = 0.0)

Public functions

void fit(DataFrame* trainData) virtual
This method creates the decision tree, which can be used later for prediction.
auto predict(DataFrame* testData) -> DataFrame* virtual

Private functions

auto computeLoss(std::vector<double> labelCounts) -> double virtual
auto getTruthVector(DataFrame*) -> std::vector<double> virtual
void printTruthVector(std::vector<double> truthVector) virtual

Function documentation

void DecisionTreeClassifier::fit(DataFrame* trainData) virtual

This method creates the decision tree, which can be used later for prediction.

Parameters
trainData