DiagGMM.h

// Copyright (C) 2001 Jean-Marc Valin

#ifndef DIAG_GMM_H
#define DIAG_GMM_H
#include "Object.h"
#include <vector>

namespace FD {

class GMM;

class DiagGMM : public Object {
   char *ptr;

   float *base;

   int dim;

   int augDim;

   int nbGauss;

public:
   //DiagGMM(const GMM &gmm);
   DiagGMM() {ptr=NULL;}

   ~DiagGMM() {if (ptr) delete [] ptr;}

   void train (const std::vector<float *> &frames, int nb_dim, int nb_gaussians, int nb_splits);

   float score(const float *vec);

   int getDim() {return dim;}

   virtual void printOn(std::ostream &out=std::cout) const;

   void readFrom (std::istream &in=std::cin);

   friend std::istream &operator >> (std::istream &in, DiagGMM &gmm);

   virtual void serialize(std::ostream &out) const;

   virtual void unserialize(std::istream &in);

   friend class GMM;
};

std::istream &operator >> (std::istream &in, DiagGMM &gmm);

}//namespace FD

#endif /* DIAG_GMM_H */

---