QuickRank  v2.0
QuickRank: A C++ suite of Learning to Rank algorithms
Public Member Functions | Protected Attributes | Private Member Functions | List of all members
RegressionTree Class Reference

#include <rt.h>

Inheritance diagram for RegressionTree:
ObliviousRT

Public Member Functions

 RegressionTree (size_t nrequiredleaves, quickrank::data::VerticalDataset *dps, double *labels, size_t minls)
 
 ~RegressionTree ()
 
void fit (RTNodeHistogram *hist)
 
double update_output (double const *pseudoresponses)
 
double update_output (double const *pseudoresponses, double const *cachedweights)
 
RTNodeget_proot () const
 

Protected Attributes

const size_t nrequiredleaves
 
const size_t minls
 
quickrank::data::VerticalDatasettraining_dataset = NULL
 
double * training_labels = NULL
 
RTNode ** leaves = NULL
 
size_t nleaves = 0
 
RTNoderoot = NULL
 

Private Member Functions

bool split (RTNode *node, const float featuresamplingrate, const bool require_devianceltparent)
 

Constructor & Destructor Documentation

RegressionTree::RegressionTree ( size_t  nrequiredleaves,
quickrank::data::VerticalDataset dps,
double *  labels,
size_t  minls 
)
inline
RegressionTree::~RegressionTree ( )
Todo:
TODO: memory management of regression tree is wrong!!!

Member Function Documentation

void RegressionTree::fit ( RTNodeHistogram hist)
RTNode* RegressionTree::get_proot ( ) const
inline
bool RegressionTree::split ( RTNode node,
const float  featuresamplingrate,
const bool  require_devianceltparent 
)
private
double RegressionTree::update_output ( double const *  pseudoresponses)
double RegressionTree::update_output ( double const *  pseudoresponses,
double const *  cachedweights 
)

Member Data Documentation

RTNode** RegressionTree::leaves = NULL
protected
const size_t RegressionTree::minls
protected
size_t RegressionTree::nleaves = 0
protected
const size_t RegressionTree::nrequiredleaves
protected
RTNode* RegressionTree::root = NULL
protected
quickrank::data::VerticalDataset* RegressionTree::training_dataset = NULL
protected
double* RegressionTree::training_labels = NULL
protected

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