conifer::BDT< T, U, useAddTree > Class Template Reference

#include <conifer.h>

Collaboration diagram for conifer::BDT< T, U, useAddTree >:

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

Private Attributes
int	m_n_classes
int	m_n_trees
int	m_n_features
std::vector< double >	m_init_predict
std::vector< U >	m_init_predict_
std::vector< std::vector< DecisionTree< T, U > > >	m_trees
OpAdd< U >	m_add

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

Detailed Description

template<class T, class U, bool useAddTree = false>
class conifer::BDT< T, U, useAddTree >

Definition at line 96 of file conifer.h.

Member Function Documentation

decision_function()

template<class T , class U , bool useAddTree = false>

std::vector<U> conifer::BDT< T, U, useAddTree >::decision_function ( std::vector< T >  x ) const

inline

Definition at line 128 of file conifer.h.

                                                       {
    /* Do the prediction */
    assert("Size of feature vector mismatches expected m_n_features" &&
           x.size() == static_cast<size_t>(m_n_features));
    std::vector<U> values;
    std::vector<std::vector<U>> values_trees;
    values_trees.resize(m_n_classes);
    values.resize(m_n_classes, U(0));
    for (int i = 0; i < m_n_classes; i++) {
      std::transform(
          m_trees.begin(), m_trees.end(),
          std::back_inserter(values_trees.at(i)),
          [&i, &x](auto tree_v) { return tree_v.at(i).decision_function(x); });
      if (useAddTree) {
        values.at(i) = m_init_predict_.at(i);
        values.at(i) += reduce<U, OpAdd<U>>(values_trees.at(i), m_add);
      } else {
        values.at(i) =
            std::accumulate(values_trees.at(i).begin(),
                            values_trees.at(i).end(), U(m_init_predict_.at(i)));
      }
    }
 
    return values;
  }

init()

template<class T , class U , bool useAddTree = false>

void conifer::BDT< T, U, useAddTree >::init ( )

inline

Definition at line 109 of file conifer.h.

               {
    /* Construct the BDT from conifer cpp backend JSON file */
    //  std::ifstream ifs(filename);
    //  nlohmann::json j = nlohmann::json::parse(ifs);
    //  from_json(j, *this);
    /* Do some transformation to initialise things into the proper emulation T,
     * U types */
    if (m_n_classes == 2)
      m_n_classes = 1;
    std::transform(m_init_predict.begin(), m_init_predict.end(),
                   std::back_inserter(m_init_predict_),
                   [](double ip) -> U { return (U)ip; });
    for (int i = 0; i < m_n_trees; i++) {
      for (int j = 0; j < m_n_classes; j++) {
        m_trees.at(i).at(j).init_();
      }
    }
  }

Friends And Related Function Documentation

from_json

template<class T , class U , bool useAddTree = false>

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

friend

Definition at line 155 of file conifer.h.

                                                       {
    j.at("n_classes").get_to(o.m_n_classes);
    j.at("n_trees").get_to(o.m_n_trees);
    j.at("n_features").get_to(o.m_n_features);
    j.at("init_predict").get_to(o.m_init_predict);
    j.at("trees").get_to(o.m_trees);
  }

Member Data Documentation

m_add

template<class T , class U , bool useAddTree = false>

OpAdd<U> conifer::BDT< T, U, useAddTree >::m_add

private

Definition at line 106 of file conifer.h.

m_init_predict

template<class T , class U , bool useAddTree = false>

std::vector<double> conifer::BDT< T, U, useAddTree >::m_init_predict

private

Definition at line 102 of file conifer.h.

m_init_predict_

template<class T , class U , bool useAddTree = false>

std::vector<U> conifer::BDT< T, U, useAddTree >::m_init_predict_

private

Definition at line 103 of file conifer.h.

m_n_classes

template<class T , class U , bool useAddTree = false>

int conifer::BDT< T, U, useAddTree >::m_n_classes

private

Definition at line 99 of file conifer.h.

m_n_features

template<class T , class U , bool useAddTree = false>

int conifer::BDT< T, U, useAddTree >::m_n_features

private

Definition at line 101 of file conifer.h.

m_n_trees

template<class T , class U , bool useAddTree = false>

int conifer::BDT< T, U, useAddTree >::m_n_trees

private

Definition at line 100 of file conifer.h.

m_trees

template<class T , class U , bool useAddTree = false>

std::vector<std::vector<DecisionTree<T, U> > > conifer::BDT< T, U, useAddTree >::m_trees

private

Definition at line 105 of file conifer.h.

---

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

• conifer.h