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

#include <conifer.h>

Public Member Functions
U	decision_function (std::vector< T > x) const
void	init_ ()
	NLOHMANN_DEFINE_TYPE_INTRUSIVE (DecisionTree, feature, children_left, children_right, threshold, value)

Private Attributes
std::vector< int >	children_left
std::vector< int >	children_right
std::vector< int >	feature
std::vector< double >	threshold
std::vector< T >	threshold_
std::vector< double >	value
std::vector< U >	value_

Detailed Description

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

Definition at line 53 of file conifer.h.

Member Function Documentation

decision_function()

template<class T , class U >

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

inline

Definition at line 64 of file conifer.h.

References conifer::DecisionTree< T, U >::children_left, conifer::DecisionTree< T, U >::children_right, conifer::DecisionTree< T, U >::feature, mps_fire::i, conifer::DecisionTree< T, U >::threshold_, conifer::DecisionTree< T, U >::value_, and x.

                                              {
      /* Do the prediction */
      int i = 0;
      while (feature[i] != -2) {  // continue until reaching leaf
        bool comparison = x[feature[i]] <= threshold_[i];
        i = comparison ? children_left[i] : children_right[i];
      }
      return value_[i];
    }

init_()

template<class T , class U >

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

inline

Definition at line 74 of file conifer.h.

References submitPVValidationJobs::t, conifer::DecisionTree< T, U >::threshold, conifer::DecisionTree< T, U >::threshold_, HcalDetIdTransform::transform(), mitigatedMETSequence_cff::U, findQualityFiles::v, and conifer::DecisionTree< T, U >::value_.

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

NLOHMANN_DEFINE_TYPE_INTRUSIVE()

template<class T , class U >

conifer::DecisionTree< T, U >::NLOHMANN_DEFINE_TYPE_INTRUSIVE	(	DecisionTree< T, U >	,
		feature	,
		children_left	,
		children_right	,
		threshold	,
		value
	)

Member Data Documentation

children_left

template<class T , class U >

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

private

Definition at line 56 of file conifer.h.

Referenced by conifer::DecisionTree< T, U >::decision_function().

children_right

template<class T , class U >

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

private

Definition at line 57 of file conifer.h.

Referenced by conifer::DecisionTree< T, U >::decision_function().

feature

template<class T , class U >

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

private

Definition at line 55 of file conifer.h.

Referenced by conifer::DecisionTree< T, U >::decision_function().

threshold

template<class T , class U >

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

private

Definition at line 60 of file conifer.h.

Referenced by utils.StatisticalTest::get_status(), and conifer::DecisionTree< T, U >::init_().

threshold_

template<class T , class U >

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

private

Definition at line 58 of file conifer.h.

Referenced by conifer::DecisionTree< T, U >::decision_function(), and conifer::DecisionTree< T, U >::init_().

value

template<class T , class U >

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

private

Definition at line 61 of file conifer.h.

Referenced by Types.int32::__nonzero__(), Types.uint32::__nonzero__(), Types.int64::__nonzero__(), Types.uint64::__nonzero__(), Types.double::__nonzero__(), Types.bool::__nonzero__(), Types.string::__nonzero__(), timeout.TimedOutExc::__str__(), dataset.IntegrityCheckError::__str__(), cmsBatch.CmsBatchException::__str__(), average.Average::average(), Types.string::configValue(), Types.FileInPath::configValue(), Mixins.UsingBlock::dumpPython(), Mixins.UsingBlock::insertInto(), Types.int32::insertInto(), Types.uint32::insertInto(), Types.int64::insertInto(), Types.uint64::insertInto(), Types.double::insertInto(), Types.bool::insertInto(), Types.string::insertInto(), Types.FileInPath::insertInto(), Types.vint32::insertInto(), Types.vuint32::insertInto(), Types.vint64::insertInto(), Types.vuint64::insertInto(), Types.vdouble::insertInto(), Types.vbool::insertInto(), and Types.vstring::insertInto().

value_

template<class T , class U >

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

private

Definition at line 59 of file conifer.h.

Referenced by conifer::DecisionTree< T, U >::decision_function(), and conifer::DecisionTree< T, U >::init_().