#include <CompiledModelInterface.hpp>

Public Member Functions
	CompiledModelInterface (int nbatch, bool is_pdvd)
	~CompiledModelInterface ()
int	GetNumFeatures ()
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,
		bool	is_pdvd )

Definition at line 7 of file CompiledModelInterface.cpp.

                                                                       : num_batch(nbatch) {
  if (is_pdvd) {  
    model_ptr = std::make_unique<TreelitePDVDModel>();
  } else {
    model_ptr = std::make_unique<TreelitePDHDModel>();
  }
}

◆ ~CompiledModelInterface()

triggeralgs::CompiledModelInterface::~CompiledModelInterface ( )

Definition at line 15 of file CompiledModelInterface.cpp.

15{}

Member Function Documentation

◆ Classify()

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

Definition at line 27 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 17 of file CompiledModelInterface.cpp.

                                           {
  return model_ptr->get_num_feature();
}

◆ Predict()

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

Definition at line 21 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 35 of file CompiledModelInterface.hpp.

◆ num_batch

int triggeralgs::CompiledModelInterface::num_batch