triggeralgs::TreeliteModelBase Class Reference

Base class: defines shared interface and utilities for all models. More...

#include <treelitemodel.hpp>

Inheritance diagram for triggeralgs::TreeliteModelBase:

Public Member Functions
	TreeliteModelBase (int numTargets=1, int maxNumClass=1)
virtual	~TreeliteModelBase ()=default
int32_t	get_num_target () const
void	get_num_class (int32_t *out) const
virtual int32_t	get_num_feature () const =0
virtual const char *	get_threshold_type () const
virtual const char *	get_leaf_output_type () const
virtual void	predict (union Entry *data, int pred_margin, float *result) const =0
virtual void	postprocess (float *result) const

Protected Attributes
int32_t	N_TARGET
int32_t	MAX_N_CLASS

Static Protected Attributes
static const int32_t	num_class [1] = { 1 }

Detailed Description

Base class: defines shared interface and utilities for all models.

Definition at line 27 of file treelitemodel.hpp.

Constructor & Destructor Documentation

TreeliteModelBase()

triggeralgs::TreeliteModelBase::TreeliteModelBase	(	int	numTargets = 1,
		int	maxNumClass = 1 )

Definition at line 10 of file treelitemodel.cpp.

  : N_TARGET(numTargets), MAX_N_CLASS(maxNumClass) {}

~TreeliteModelBase()

virtual triggeralgs::TreeliteModelBase::~TreeliteModelBase ( )

virtualdefault

Member Function Documentation

get_leaf_output_type()

const char * triggeralgs::TreeliteModelBase::get_leaf_output_type ( void ) const

virtual

Definition at line 26 of file treelitemodel.cpp.

                                                              {
  return "float32";
}

get_num_class()

void triggeralgs::TreeliteModelBase::get_num_class

(

int32_t *

out

)

const

Definition at line 17 of file treelitemodel.cpp.

                                                        {
  for (int i = 0; i < N_TARGET; ++i) {
    out[i] = TreeliteModelBase::num_class[i];
  }
}

get_num_feature()

virtual int32_t triggeralgs::TreeliteModelBase::get_num_feature ( ) const

pure virtual

Implemented in triggeralgs::TreelitePDHDModel.

get_num_target()

int32_t triggeralgs::TreeliteModelBase::get_num_target

(

void

)

const

Definition at line 13 of file treelitemodel.cpp.

                                                    {
  return N_TARGET;
}

get_threshold_type()

const char * triggeralgs::TreeliteModelBase::get_threshold_type ( void ) const

virtual

Definition at line 23 of file treelitemodel.cpp.

                                                            {
  return "float32";
}

postprocess()

void triggeralgs::TreeliteModelBase::postprocess ( float * result ) const

virtual

Definition at line 30 of file treelitemodel.cpp.

                                                       {
  // sigmoid
  const float alpha = (float)1;
  for (size_t i = 0; i < N_TARGET * MAX_N_CLASS; ++i) {
    result[i] = (float)(1) / ((float)(1) + expf(-alpha * result[i]));
  }
}

predict()

virtual void triggeralgs::TreeliteModelBase::predict ( union Entry * data, int pred_margin, float * result ) const

pure virtual

Implemented in triggeralgs::TreelitePDHDModel.

Member Data Documentation

MAX_N_CLASS

int32_t triggeralgs::TreeliteModelBase::MAX_N_CLASS

protected

Definition at line 50 of file treelitemodel.hpp.

N_TARGET

int32_t triggeralgs::TreeliteModelBase::N_TARGET

protected

Definition at line 49 of file treelitemodel.hpp.

num_class

const int32_t triggeralgs::TreeliteModelBase::num_class = { 1 }

staticprotected

Definition at line 48 of file treelitemodel.hpp.

---

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

• /github/workspace/dunedaq/sourcecode/triggeralgs/include/triggeralgs/ProtoDUNEBSMWindow/treelitemodel.hpp
• /github/workspace/dunedaq/sourcecode/triggeralgs/src/ProtoDUNEBSMWindow/treelitemodel.cpp