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