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

#include <conifer.h>

Public Member Functions
std::vector< double >	_decision_function_double (std::vector< double > x) const
	BDT (std::string filename)
std::vector< U >	decision_function (std::vector< T > x) const
	NLOHMANN_DEFINE_TYPE_INTRUSIVE (BDT, n_classes, n_trees, n_features, init_predict, trees)

Private Attributes
OpAdd< U >	add
std::vector< double >	init_predict
std::vector< U >	init_predict_
unsigned int	n_classes
unsigned int	n_features
unsigned int	n_trees
std::vector< std::vector< DecisionTree< T, U > > >	trees

Detailed Description

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

Definition at line 87 of file conifer.h.

Constructor & Destructor Documentation

BDT()

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

conifer::BDT< T, U, useAddTree >::BDT ( std::string  filename )

inline

Definition at line 102 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 (n_classes == 2)
        n_classes = 1;
      std::transform(init_predict.begin(), init_predict.end(), std::back_inserter(init_predict_), [](double ip) -> U {
        return (U)ip;
      });
      for (unsigned int i = 0; i < n_trees; i++) {
        for (unsigned int j = 0; j < n_classes; j++) {
          trees.at(i).at(j).init_();
        }
      }
    }

Member Function Documentation

_decision_function_double()

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

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

inline

Definition at line 152 of file conifer.h.

                                                                           {
      /* Do the prediction with data in/out as double, cast to T, U before prediction */
      std::vector<T> xt;
      std::transform(x.begin(), x.end(), std::back_inserter(xt), [](double xi) -> T { return (T)xi; });
      std::vector<U> y = decision_function(xt);
      std::vector<double> yd;
      std::transform(y.begin(), y.end(), std::back_inserter(yd), [](U yi) -> double { return (double)yi; });
      return yd;
    }

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 120 of file conifer.h.

Referenced by conifer::BDT< bdt_feature_t, ap_fixed< 12, 3, AP_RND_CONV, AP_SAT >, false >::_decision_function_double().

                                                         {
      /* Do the prediction */
#ifdef CMSSW_GIT_HASH
      if (x.size() != n_features) {
        throw cms::Exception("RuntimeError")
            << "Conifer : Size of feature vector mismatches expected n_features" << std::endl;
      }
#else
      if (x.size() != n_features) {
        throw std::runtime_error("Conifer : Size of feature vector mismatches expected n_features");
      }
#endif
      std::vector<U> values;
      std::vector<std::vector<U>> values_trees;
      values_trees.resize(n_classes);
      values.resize(n_classes, U(0));
      for (unsigned int i = 0; i < n_classes; i++) {
        std::transform(trees.begin(),
                       trees.end(),
                       std::back_inserter(values_trees.at(i)),
                       [&i, &x](std::vector<DecisionTree<T, U>> tree_v) { return tree_v.at(i).decision_function(x); });
        if (useAddTree) {
          values.at(i) = init_predict_.at(i);
          values.at(i) += reduce<U, OpAdd<U>>(values_trees.at(i), add);
        } else {
          values.at(i) = std::accumulate(values_trees.at(i).begin(), values_trees.at(i).end(), U(init_predict_.at(i)));
        }
      }

      return values;
    }

NLOHMANN_DEFINE_TYPE_INTRUSIVE()

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

conifer::BDT< T, U, useAddTree >::NLOHMANN_DEFINE_TYPE_INTRUSIVE	(	BDT< T, U, useAddTree >	,
		n_classes	,
		n_trees	,
		n_features	,
		init_predict	,
		trees
	)

Member Data Documentation

add

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

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

private

Definition at line 96 of file conifer.h.

Referenced by conifer::BDT< bdt_feature_t, ap_fixed< 12, 3, AP_RND_CONV, AP_SAT >, false >::decision_function(), counter.Counter::register(), SequenceTypes._TaskBase::remove(), and SequenceTypes._TaskBase::replace().

init_predict

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

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

private

Definition at line 92 of file conifer.h.

Referenced by conifer::BDT< bdt_feature_t, ap_fixed< 12, 3, AP_RND_CONV, AP_SAT >, false >::BDT().

init_predict_

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

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

private

Definition at line 93 of file conifer.h.

Referenced by conifer::BDT< bdt_feature_t, ap_fixed< 12, 3, AP_RND_CONV, AP_SAT >, false >::BDT(), and conifer::BDT< bdt_feature_t, ap_fixed< 12, 3, AP_RND_CONV, AP_SAT >, false >::decision_function().

n_classes

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

unsigned int conifer::BDT< T, U, useAddTree >::n_classes

private

Definition at line 89 of file conifer.h.

Referenced by conifer::BDT< bdt_feature_t, ap_fixed< 12, 3, AP_RND_CONV, AP_SAT >, false >::BDT(), and conifer::BDT< bdt_feature_t, ap_fixed< 12, 3, AP_RND_CONV, AP_SAT >, false >::decision_function().

n_features

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

unsigned int conifer::BDT< T, U, useAddTree >::n_features

private

Definition at line 91 of file conifer.h.

Referenced by conifer::BDT< bdt_feature_t, ap_fixed< 12, 3, AP_RND_CONV, AP_SAT >, false >::decision_function().

n_trees

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

unsigned int conifer::BDT< T, U, useAddTree >::n_trees

private

Definition at line 90 of file conifer.h.

Referenced by conifer::BDT< bdt_feature_t, ap_fixed< 12, 3, AP_RND_CONV, AP_SAT >, false >::BDT().

trees

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

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

private

Definition at line 95 of file conifer.h.

Referenced by conifer::BDT< bdt_feature_t, ap_fixed< 12, 3, AP_RND_CONV, AP_SAT >, false >::BDT(), and conifer::BDT< bdt_feature_t, ap_fixed< 12, 3, AP_RND_CONV, AP_SAT >, false >::decision_function().