model-common.h

Go to the documentation of this file.

// gmm/model-common.h

// Copyright 2009-2012  Saarland University;  Microsoft Corporation;
//                      Johns Hopkins University (author: Daniel Povey)

// See ../../COPYING for clarification regarding multiple authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//  http://www.apache.org/licenses/LICENSE-2.0
//
// THIS CODE IS PROVIDED *AS IS* BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, EITHER EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION ANY IMPLIED
// WARRANTIES OR CONDITIONS OF TITLE, FITNESS FOR A PARTICULAR PURPOSE,
// MERCHANTABLITY OR NON-INFRINGEMENT.
// See the Apache 2 License for the specific language governing permissions and
// limitations under the License.


#ifndef KALDI_GMM_MODEL_COMMON_H_
#define KALDI_GMM_MODEL_COMMON_H_
#include "matrix/matrix-lib.h"

namespace kaldi {

enum GmmUpdateFlags {
  kGmmMeans       = 0x001,  // m
  kGmmVariances   = 0x002,  // v
  kGmmWeights     = 0x004,  // w
  kGmmTransitions = 0x008,  // t ... not really part of GMM.
  kGmmAll       = 0x00F  // a
};
typedef uint16 GmmFlagsType;  
GmmFlagsType StringToGmmFlags(std::string str);

std::string GmmFlagsToString(GmmFlagsType gmm_flags);

// Make sure that the flags make sense, i.e. if there is variance
// accumulation that there is also mean accumulation
GmmFlagsType AugmentGmmFlags(GmmFlagsType flags);

enum SgmmUpdateFlags {  
  kSgmmPhoneVectors                = 0x001,  
  kSgmmPhoneProjections            = 0x002,  
  kSgmmPhoneWeightProjections      = 0x004,  
  kSgmmCovarianceMatrix            = 0x008,  
  kSgmmSubstateWeights             = 0x010,  
  kSgmmSpeakerProjections          = 0x020,  
  kSgmmTransitions                 = 0x040,  
  kSgmmSpeakerWeightProjections    = 0x080,  
  kSgmmAll                         = 0x0FF   
};

typedef uint16 SgmmUpdateFlagsType;  
SgmmUpdateFlagsType StringToSgmmUpdateFlags(std::string str);

enum SgmmWriteFlags {
  kSgmmGlobalParams    = 0x001,  
  kSgmmStateParams     = 0x002,  
  kSgmmNormalizers     = 0x004,  
  kSgmmBackgroundGmms  = 0x008,  
  kSgmmWriteAll        = 0x00F  
};

typedef uint16 SgmmWriteFlagsType;  

SgmmWriteFlagsType StringToSgmmWriteFlags(std::string str);

void GetSplitTargets(const Vector<BaseFloat> &state_occs,
                     int32 target_components,
                     BaseFloat power,
                     BaseFloat min_count,
                     std::vector<int32> *targets);

}  // End namespace kaldi


#endif  // KALDI_GMM_MODEL_COMMON_H_