mog_diag_kmeans.h

Go to the documentation of this file.

 
 
#ifndef MOG_DIAG_KMEANS_H
#define MOG_DIAG_KMEANS_H
 
#include <itpp/stat/mog_diag.h>
#include <itpp/itexports.h>
#include <itpp/base/base_exports.h>
 
namespace itpp
{

class ITPP_EXPORT MOG_diag_kmeans_sup : public MOG_diag
{
 
public:
  MOG_diag_kmeans_sup() { verbose = false; }

  ~MOG_diag_kmeans_sup() { }

  void run(MOG_diag &model_in, Array<vec> &X_in, int max_iter_in = 10, double trust_in = 0.5, bool normalise_in = true, bool verbose_in = false);
 
protected:

  inline double dist(const double * x, const double * y) const;
  void assign_to_means();
  void recalculate_means();
  bool dezombify_means();
  double measure_change() const;
  void initial_means();
  void iterate();
  void calc_means();
  void calc_covs();
  void calc_weights();
  void normalise_vectors();
  void unnormalise_vectors();
  void unnormalise_means();

  int max_iter;

  double trust;

  bool verbose;

  int N;

  double ** c_X;

  Array<vec> means_old;

  double ** c_means_old;

  Array<ivec> partitions;

  int ** c_partitions;

  ivec count;

  int * c_count;
 
private:
 
  vec norm_mu;
  double * c_norm_mu;
 
  vec norm_sd;
  double * c_norm_sd;
 
  vec tmpvec;
  double * c_tmpvec;
 
 
};
 
//
// convenience functions

void MOG_diag_kmeans(MOG_diag &model_in, Array<vec> &X_in, int max_iter_in = 10, double trust_in = 0.5, bool normalise_in = true, bool verbose_in = false);
 
}
 
#endif  // #ifndef MOG_DIAG_KMEANS_H