CvEM Class Reference

#include <ml.h>

Inherits CvStatModel.

List of all members.

Public Types

enum  { COV_MAT_SPHERICAL = 0, COV_MAT_DIAGONAL = 1, COV_MAT_GENERIC = 2 }
enum  { START_E_STEP = 1, START_M_STEP = 2, START_AUTO_STEP = 0 }

Public Member Functions

 CvEM ()
 CvEM (const CvMat *samples, const CvMat *sample_idx=0, CvEMParams params=CvEMParams(), CvMat *labels=0)
virtual ~CvEM ()
virtual bool train (const CvMat *samples, const CvMat *sample_idx=0, CvEMParams params=CvEMParams(), CvMat *labels=0)
virtual float predict (const CvMat *sample, CvMat *probs) const
 CvEM (const cv::Mat &samples, const cv::Mat &sample_idx=cv::Mat(), CvEMParams params=CvEMParams(), cv::Mat *labels=0)
virtual bool train (const cv::Mat &samples, const cv::Mat &sample_idx=cv::Mat(), CvEMParams params=CvEMParams(), cv::Mat *labels=0)
virtual float predict (const cv::Mat &sample, cv::Mat *probs) const
virtual void clear ()
int get_nclusters () const
const CvMatget_means () const
const CvMat ** get_covs () const
const CvMatget_weights () const
const CvMatget_probs () const
double get_log_likelihood () const
virtual void save (const char *filename, const char *name=0) const
virtual void load (const char *filename, const char *name=0)
virtual void write (CvFileStorage *storage, const char *name) const
virtual void read (CvFileStorage *storage, CvFileNode *node)

Protected Member Functions

virtual void set_params (const CvEMParams &params, const CvVectors &train_data)
virtual void init_em (const CvVectors &train_data)
virtual double run_em (const CvVectors &train_data)
virtual void init_auto (const CvVectors &samples)
virtual void kmeans (const CvVectors &train_data, int nclusters, CvMat *labels, CvTermCriteria criteria, const CvMat *means)

Protected Attributes

CvEMParams params
double log_likelihood
CvMatmeans
CvMat ** covs
CvMatweights
CvMatprobs
CvMatlog_weight_div_det
CvMatinv_eigen_values
CvMat ** cov_rotate_mats
const char * default_model_name

Member Enumeration Documentation

anonymous enum
Enumerator:
COV_MAT_SPHERICAL 
COV_MAT_DIAGONAL 
COV_MAT_GENERIC 
anonymous enum
Enumerator:
START_E_STEP 
START_M_STEP 
START_AUTO_STEP 

Constructor & Destructor Documentation

CvEM::CvEM (  ) 
CvEM::CvEM ( const CvMat samples,
const CvMat sample_idx = 0,
CvEMParams  params = CvEMParams(),
CvMat labels = 0 
)
virtual CvEM::~CvEM (  )  [virtual]
CvEM::CvEM ( const cv::Mat samples,
const cv::Mat sample_idx = cv::Mat(),
CvEMParams  params = CvEMParams(),
cv::Mat labels = 0 
)

Member Function Documentation

virtual bool CvEM::train ( const CvMat samples,
const CvMat sample_idx = 0,
CvEMParams  params = CvEMParams(),
CvMat labels = 0 
) [virtual]
virtual float CvEM::predict ( const CvMat sample,
CvMat probs 
) const [virtual]
virtual bool CvEM::train ( const cv::Mat samples,
const cv::Mat sample_idx = cv::Mat(),
CvEMParams  params = CvEMParams(),
cv::Mat labels = 0 
) [virtual]
virtual float CvEM::predict ( const cv::Mat sample,
cv::Mat probs 
) const [virtual]
virtual void CvEM::clear (  )  [virtual]

Reimplemented from CvStatModel.

int CvEM::get_nclusters (  )  const
const CvMat* CvEM::get_means (  )  const
const CvMat** CvEM::get_covs (  )  const
const CvMat* CvEM::get_weights (  )  const
const CvMat* CvEM::get_probs (  )  const
double CvEM::get_log_likelihood (  )  const
virtual void CvEM::set_params ( const CvEMParams params,
const CvVectors train_data 
) [protected, virtual]
virtual void CvEM::init_em ( const CvVectors train_data  )  [protected, virtual]
virtual double CvEM::run_em ( const CvVectors train_data  )  [protected, virtual]
virtual void CvEM::init_auto ( const CvVectors samples  )  [protected, virtual]
virtual void CvEM::kmeans ( const CvVectors train_data,
int  nclusters,
CvMat labels,
CvTermCriteria  criteria,
const CvMat means 
) [protected, virtual]
virtual void CvStatModel::save ( const char *  filename,
const char *  name = 0 
) const [virtual, inherited]
virtual void CvStatModel::load ( const char *  filename,
const char *  name = 0 
) [virtual, inherited]
virtual void CvStatModel::write ( CvFileStorage storage,
const char *  name 
) const [virtual, inherited]
virtual void CvStatModel::read ( CvFileStorage storage,
CvFileNode node 
) [virtual, inherited]

Member Data Documentation

CvEMParams CvEM::params [protected]
double CvEM::log_likelihood [protected]
CvMat* CvEM::means [protected]
CvMat** CvEM::covs [protected]
CvMat* CvEM::weights [protected]
CvMat* CvEM::probs [protected]
const char* CvStatModel::default_model_name [protected, inherited]

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