conifer::DecisionTree< T, U > Class Template Reference

#include <conifer.h>

Collaboration diagram for conifer::DecisionTree< T, U >:

Public Member Functions
U	decision_function (const std::vector< T > &x) const
void	init_ ()

Private Attributes
std::vector< int >	m_feature
std::vector< int >	m_children_left
std::vector< int >	m_children_right
std::vector< T >	m_threshold_
std::vector< U >	m_value_
std::vector< double >	m_threshold
std::vector< double >	m_value

Friends
void	from_json (const nlohmann::json &j, DecisionTree &o)

Detailed Description

template<class T, class U>
class conifer::DecisionTree< T, U >

Definition at line 46 of file conifer.h.

Member Function Documentation

decision_function()

template<class T , class U >

U conifer::DecisionTree< T, U >::decision_function ( const std::vector< T > &  x ) const

inline

Definition at line 58 of file conifer.h.

                                                   {
    /* Do the prediction */
    int i = 0;
    while (m_feature[i] != -2) { // continue until reaching leaf
      bool comparison = x[m_feature[i]] <= m_threshold_[i];
      i = comparison ? m_children_left[i] : m_children_right[i];
    }
    return m_value_[i];
  }

init_()

template<class T , class U >

void conifer::DecisionTree< T, U >::init_ ( )

inline

Definition at line 68 of file conifer.h.

               {
    /* Since T, U types may not be readable from the JSON, read them to double
     * and the cast them here */
    std::transform(m_threshold.begin(), m_threshold.end(),
                   std::back_inserter(m_threshold_),
                   [](double t) -> T { return (T)t; });
    std::transform(m_value.begin(), m_value.end(), std::back_inserter(m_value_),
                   [](double v) -> U { return (U)v; });
  }

Friends And Related Function Documentation

from_json

template<class T , class U >

void from_json ( const nlohmann::json &  j, DecisionTree< T, U > &  o  )

friend

Definition at line 79 of file conifer.h.

                                                                {
    j.at("feature").get_to(o.m_feature);
    j.at("children_left").get_to(o.m_children_left);
    j.at("children_right").get_to(o.m_children_right);
    j.at("threshold").get_to(o.m_threshold);
    j.at("value").get_to(o.m_value);
  }

Member Data Documentation

m_children_left

template<class T , class U >

std::vector<int> conifer::DecisionTree< T, U >::m_children_left

private

Definition at line 50 of file conifer.h.

m_children_right

template<class T , class U >

std::vector<int> conifer::DecisionTree< T, U >::m_children_right

private

Definition at line 51 of file conifer.h.

m_feature

template<class T , class U >

std::vector<int> conifer::DecisionTree< T, U >::m_feature

private

Definition at line 49 of file conifer.h.

m_threshold

template<class T , class U >

std::vector<double> conifer::DecisionTree< T, U >::m_threshold

private

Definition at line 54 of file conifer.h.

m_threshold_

template<class T , class U >

std::vector<T> conifer::DecisionTree< T, U >::m_threshold_

private

Definition at line 52 of file conifer.h.

m_value

template<class T , class U >

std::vector<double> conifer::DecisionTree< T, U >::m_value

private

Definition at line 55 of file conifer.h.

m_value_

template<class T , class U >

std::vector<U> conifer::DecisionTree< T, U >::m_value_

private

Definition at line 53 of file conifer.h.

---

The documentation for this class was generated from the following file:

• conifer.h