Public Types | Public Member Functions | Protected Member Functions | Protected Attributes
CvGBTrees Class Reference

#include <ml.hpp>

Inherits CvStatModel.

List of all members.

Public Types

enum  { SQUARED_LOSS = 0, ABSOLUTE_LOSS, HUBER_LOSS = 3, DEVIANCE_LOSS }

Public Member Functions

CV_WRAP CvGBTrees ()
 CvGBTrees (const CvMat *trainData, int tflag, const CvMat *responses, const CvMat *varIdx=0, const CvMat *sampleIdx=0, const CvMat *varType=0, const CvMat *missingDataMask=0, CvGBTreesParams params=CvGBTreesParams())
virtual ~CvGBTrees ()
virtual bool train (const CvMat *trainData, int tflag, const CvMat *responses, const CvMat *varIdx=0, const CvMat *sampleIdx=0, const CvMat *varType=0, const CvMat *missingDataMask=0, CvGBTreesParams params=CvGBTreesParams(), bool update=false)
virtual bool train (CvMLData *data, CvGBTreesParams params=CvGBTreesParams(), bool update=false)
virtual float predict_serial (const CvMat *sample, const CvMat *missing=0, CvMat *weakResponses=0, CvSlice slice=CV_WHOLE_SEQ, int k=-1) const
virtual float predict (const CvMat *sample, const CvMat *missing=0, CvMat *weakResponses=0, CvSlice slice=CV_WHOLE_SEQ, int k=-1) const
virtual CV_WRAP void clear ()
virtual float calc_error (CvMLData *_data, int type, std::vector< float > *resp=0)
virtual void write (CvFileStorage *fs, const char *name) const
virtual void read (CvFileStorage *fs, CvFileNode *node)
CV_WRAP CvGBTrees (const cv::Mat &trainData, int tflag, const cv::Mat &responses, const cv::Mat &varIdx=cv::Mat(), const cv::Mat &sampleIdx=cv::Mat(), const cv::Mat &varType=cv::Mat(), const cv::Mat &missingDataMask=cv::Mat(), CvGBTreesParams params=CvGBTreesParams())
virtual CV_WRAP bool train (const cv::Mat &trainData, int tflag, const cv::Mat &responses, const cv::Mat &varIdx=cv::Mat(), const cv::Mat &sampleIdx=cv::Mat(), const cv::Mat &varType=cv::Mat(), const cv::Mat &missingDataMask=cv::Mat(), CvGBTreesParams params=CvGBTreesParams(), bool update=false)
virtual CV_WRAP float predict (const cv::Mat &sample, const cv::Mat &missing=cv::Mat(), const cv::Range &slice=cv::Range::all(), int k=-1) const
virtual CV_WRAP void save (const char *filename, const char *name=0) const
virtual CV_WRAP void load (const char *filename, const char *name=0)

Protected Member Functions

virtual void find_gradient (const int k=0)
virtual void change_values (CvDTree *tree, const int k=0)
virtual float find_optimal_value (const CvMat *_Idx)
virtual void do_subsample ()
void leaves_get (CvDTreeNode **leaves, int &count, CvDTreeNode *node)
CvDTreeNode ** GetLeaves (const CvDTree *dtree, int &len)
virtual bool problem_type () const
virtual void write_params (CvFileStorage *fs) const
virtual void read_params (CvFileStorage *fs, CvFileNode *fnode)
int get_len (const CvMat *mat) const

Protected Attributes

CvDTreeTrainDatadata
CvGBTreesParams params
CvSeq ** weak
CvMatorig_response
CvMatsum_response
CvMatsum_response_tmp
CvMatsample_idx
CvMatsubsample_train
CvMatsubsample_test
CvMatmissing
CvMatclass_labels
cv::RNGrng
int class_count
float delta
float base_value
const char * default_model_name

Member Enumeration Documentation

anonymous enum
Enumerator:
SQUARED_LOSS 
ABSOLUTE_LOSS 
HUBER_LOSS 
DEVIANCE_LOSS 

Constructor & Destructor Documentation

CV_WRAP CvGBTrees::CvGBTrees ( )
CvGBTrees::CvGBTrees ( const CvMat trainData,
int  tflag,
const CvMat responses,
const CvMat varIdx = 0,
const CvMat sampleIdx = 0,
const CvMat varType = 0,
const CvMat missingDataMask = 0,
CvGBTreesParams  params = CvGBTreesParams() 
)
virtual CvGBTrees::~CvGBTrees ( ) [virtual]
CV_WRAP CvGBTrees::CvGBTrees ( const cv::Mat trainData,
int  tflag,
const cv::Mat responses,
const cv::Mat varIdx = cv::Mat(),
const cv::Mat sampleIdx = cv::Mat(),
const cv::Mat varType = cv::Mat(),
const cv::Mat missingDataMask = cv::Mat(),
CvGBTreesParams  params = CvGBTreesParams() 
)

Member Function Documentation

virtual bool CvGBTrees::train ( const CvMat trainData,
int  tflag,
const CvMat responses,
const CvMat varIdx = 0,
const CvMat sampleIdx = 0,
const CvMat varType = 0,
const CvMat missingDataMask = 0,
CvGBTreesParams  params = CvGBTreesParams(),
bool  update = false 
) [virtual]
virtual bool CvGBTrees::train ( CvMLData data,
CvGBTreesParams  params = CvGBTreesParams(),
bool  update = false 
) [virtual]
virtual float CvGBTrees::predict_serial ( const CvMat sample,
const CvMat missing = 0,
CvMat weakResponses = 0,
CvSlice  slice = CV_WHOLE_SEQ,
int  k = -1 
) const [virtual]
virtual float CvGBTrees::predict ( const CvMat sample,
const CvMat missing = 0,
CvMat weakResponses = 0,
CvSlice  slice = CV_WHOLE_SEQ,
int  k = -1 
) const [virtual]
virtual CV_WRAP void CvGBTrees::clear ( ) [virtual]

Reimplemented from CvStatModel.

virtual float CvGBTrees::calc_error ( CvMLData _data,
int  type,
std::vector< float > *  resp = 0 
) [virtual]
virtual void CvGBTrees::write ( CvFileStorage fs,
const char *  name 
) const [virtual]

Reimplemented from CvStatModel.

virtual void CvGBTrees::read ( CvFileStorage fs,
CvFileNode node 
) [virtual]

Reimplemented from CvStatModel.

virtual CV_WRAP bool CvGBTrees::train ( const cv::Mat trainData,
int  tflag,
const cv::Mat responses,
const cv::Mat varIdx = cv::Mat(),
const cv::Mat sampleIdx = cv::Mat(),
const cv::Mat varType = cv::Mat(),
const cv::Mat missingDataMask = cv::Mat(),
CvGBTreesParams  params = CvGBTreesParams(),
bool  update = false 
) [virtual]
virtual CV_WRAP float CvGBTrees::predict ( const cv::Mat sample,
const cv::Mat missing = cv::Mat(),
const cv::Range slice = cv::Range::all(),
int  k = -1 
) const [virtual]
virtual void CvGBTrees::find_gradient ( const int  k = 0) [protected, virtual]
virtual void CvGBTrees::change_values ( CvDTree tree,
const int  k = 0 
) [protected, virtual]
virtual float CvGBTrees::find_optimal_value ( const CvMat _Idx) [protected, virtual]
virtual void CvGBTrees::do_subsample ( ) [protected, virtual]
void CvGBTrees::leaves_get ( CvDTreeNode **  leaves,
int &  count,
CvDTreeNode node 
) [protected]
CvDTreeNode** CvGBTrees::GetLeaves ( const CvDTree dtree,
int &  len 
) [protected]
virtual bool CvGBTrees::problem_type ( ) const [protected, virtual]
virtual void CvGBTrees::write_params ( CvFileStorage fs) const [protected, virtual]
virtual void CvGBTrees::read_params ( CvFileStorage fs,
CvFileNode fnode 
) [protected, virtual]
int CvGBTrees::get_len ( const CvMat mat) const [protected]
virtual CV_WRAP void CvStatModel::save ( const char *  filename,
const char *  name = 0 
) const [virtual, inherited]
virtual CV_WRAP void CvStatModel::load ( const char *  filename,
const char *  name = 0 
) [virtual, inherited]

Member Data Documentation

CvSeq** CvGBTrees::weak [protected]
CvMat* CvGBTrees::missing [protected]
cv::RNG* CvGBTrees::rng [protected]
int CvGBTrees::class_count [protected]
float CvGBTrees::delta [protected]
float CvGBTrees::base_value [protected]
const char* CvStatModel::default_model_name [protected, inherited]

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