#include <CompiledModelInterface.hpp>

Public Member Functions
	CompiledModelInterface (int nbatch)
	~CompiledModelInterface ()
int	GetNumFeatures ()
void	ModelWarmUp (Entry *input)
void	Predict (Entry input, float result)
bool	Classify (const float *result, float &bdt_threshold)

Protected Attributes
std::unique_ptr< TreeliteModelBase >	model_ptr
int	num_batch

Detailed Description

Definition at line 17 of file CompiledModelInterface.hpp.

Constructor & Destructor Documentation

◆ CompiledModelInterface()

triggeralgs::CompiledModelInterface::CompiledModelInterface ( int nbatch )

Definition at line 7 of file CompiledModelInterface.cpp.

                                                         : num_batch(nbatch) {
  model_ptr = std::make_unique<TreelitePDHDModel>();
}

◆ ~CompiledModelInterface()

triggeralgs::CompiledModelInterface::~CompiledModelInterface ( )

Definition at line 11 of file CompiledModelInterface.cpp.

11{}

Member Function Documentation

◆ Classify()

bool triggeralgs::CompiledModelInterface::Classify	(	const float *	result,
		float &	bdt_threshold )

Definition at line 33 of file CompiledModelInterface.cpp.

                                                                               {
  for (uint64_t rid = 0; rid < num_batch; rid++) {
    if (result[rid] > bdt_threshold) {
      return true;
    }
  }
  return false;
}

◆ GetNumFeatures()

int triggeralgs::CompiledModelInterface::GetNumFeatures ( )

Definition at line 13 of file CompiledModelInterface.cpp.

                                           {
  return model_ptr->get_num_feature();
}

◆ ModelWarmUp()

void triggeralgs::CompiledModelInterface::ModelWarmUp ( Entry * input )

Definition at line 17 of file CompiledModelInterface.cpp.

                                                     {
  // Warm the BDT up here
  float result[num_batch];
  for (int rid = 0; rid < num_batch; ++rid) {
    for (int i = 0; i < 100; i++) {
      model_ptr->predict(input, 0, result);
    }
  }
}

◆ Predict()

void triggeralgs::CompiledModelInterface::Predict	(	Entry *	input,
		float *	result )

Definition at line 27 of file CompiledModelInterface.cpp.

                                                                {
  for (int rid = 0; rid < num_batch; ++rid) {
    model_ptr->predict(input, 0, result);
  }
}

Member Data Documentation

◆ model_ptr

std::unique_ptr<TreeliteModelBase> triggeralgs::CompiledModelInterface::model_ptr

protected

Definition at line 37 of file CompiledModelInterface.hpp.

◆ num_batch

int triggeralgs::CompiledModelInterface::num_batch