Public Member Functions | Public Attributes | List of all members
DiscriminativeSupervision Struct Reference

#include <discriminative-supervision.h>

Collaboration diagram for DiscriminativeSupervision:
Collaboration graph
[legend]

Public Member Functions

 DiscriminativeSupervision ()
 
 DiscriminativeSupervision (const DiscriminativeSupervision &other)
 
bool Initialize (const std::vector< int32 > &alignment, const Lattice &lat, BaseFloat weight)
 
void Swap (DiscriminativeSupervision *other)
 
bool operator== (const DiscriminativeSupervision &other) const
 
void Check () const
 
int32 NumFrames () const
 
void Write (std::ostream &os, bool binary) const
 
void Read (std::istream &is, bool binary)
 

Public Attributes

BaseFloat weight
 
int32 num_sequences
 
int32 frames_per_sequence
 
std::vector< int32 > num_ali
 
Lattice den_lat
 

Detailed Description

Definition at line 75 of file discriminative-supervision.h.

Constructor & Destructor Documentation

◆ DiscriminativeSupervision() [1/2]

DiscriminativeSupervision ( )
inline

Definition at line 106 of file discriminative-supervision.h.

References fst::operator==().

106  : weight(1.0), num_sequences(1),
107  frames_per_sequence(-1) { }

◆ DiscriminativeSupervision() [2/2]

DiscriminativeSupervision ( const DiscriminativeSupervision &  other)

Definition at line 28 of file discriminative-supervision.cc.

29  :
30  weight(other.weight), num_sequences(other.num_sequences),
31  frames_per_sequence(other.frames_per_sequence),
32  num_ali(other.num_ali), den_lat(other.den_lat) { }

Member Function Documentation

◆ Check()

void Check ( ) const

Definition at line 124 of file discriminative-supervision.cc.

References DiscriminativeSupervision::den_lat, DiscriminativeSupervision::frames_per_sequence, KALDI_ASSERT, kaldi::LatticeStateTimes(), DiscriminativeSupervision::num_ali, and DiscriminativeSupervision::num_sequences.

Referenced by DiscriminativeSupervisionSplitter::GetFrameRange(), DiscriminativeSupervision::Initialize(), and kaldi::discriminative::MergeSupervision().

124  {
125  int32 num_frames_subsampled = num_ali.size();
126  KALDI_ASSERT(num_frames_subsampled ==
127  num_sequences * frames_per_sequence);
128 
129  {
130  std::vector<int32> state_times;
131  int32 max_time = LatticeStateTimes(den_lat, &state_times);
132  KALDI_ASSERT(max_time == num_frames_subsampled);
133  }
134 }
kaldi::LatticeStateTimes
int32 LatticeStateTimes(const Lattice &lat, vector< int32 > *times)
This function iterates over the states of a topologically sorted lattice and counts the time instance...
Definition: lattice-functions.cc:78
kaldi::int32
kaldi::int32 int32
Definition: online-tcp-source.cc:27
KALDI_ASSERT
#define KALDI_ASSERT(cond)
Definition: kaldi-error.h:185

◆ Initialize()

bool Initialize ( const std::vector< int32 > &  alignment,
const Lattice &  lat,
BaseFloat  weight 
)

Definition at line 105 of file discriminative-supervision.cc.

References DiscriminativeSupervision::Check(), DiscriminativeSupervision::den_lat, DiscriminativeSupervision::frames_per_sequence, KALDI_ASSERT, DiscriminativeSupervision::num_ali, DiscriminativeSupervision::num_sequences, and DiscriminativeSupervision::weight.

Referenced by main().

107  {
108  if (num_ali.size() == 0) return false;
109  if (den_lat.NumStates() == 0) return false;
110 
111  this->weight = weight;
112  this->num_sequences = 1;
113  this->frames_per_sequence = num_ali.size();
114  this->num_ali = num_ali;
115  this->den_lat = den_lat;
116  KALDI_ASSERT(TopSort(&(this->den_lat)));
117 
118  // Checks if num frames in alignment matches lattice
119  Check();
120 
121  return true;
122 }
KALDI_ASSERT
#define KALDI_ASSERT(cond)
Definition: kaldi-error.h:185

◆ NumFrames()

int32 NumFrames ( ) const
inline

Definition at line 129 of file discriminative-supervision.h.

129  {
130  return num_sequences * frames_per_sequence;
131  }

◆ operator==()

bool operator== ( const DiscriminativeSupervision &  other) const

Definition at line 42 of file discriminative-supervision.cc.

References DiscriminativeSupervision::den_lat, DiscriminativeSupervision::frames_per_sequence, DiscriminativeSupervision::num_ali, DiscriminativeSupervision::num_sequences, and DiscriminativeSupervision::weight.

43  {
44  return ( weight == other.weight &&
45  num_sequences == other.num_sequences &&
46  frames_per_sequence == other.frames_per_sequence &&
47  num_ali == other.num_ali &&
48  fst::Equal(den_lat, other.den_lat) );
49 }

◆ Read()

void Read ( std::istream &  is,
bool  binary 
)

Definition at line 75 of file discriminative-supervision.cc.

References DiscriminativeSupervision::den_lat, kaldi::ExpectToken(), DiscriminativeSupervision::frames_per_sequence, KALDI_ASSERT, KALDI_ERR, DiscriminativeSupervision::num_ali, DiscriminativeSupervision::num_sequences, kaldi::ReadBasicType(), kaldi::ReadIntegerVector(), kaldi::ReadLattice(), and DiscriminativeSupervision::weight.

Referenced by NnetDiscriminativeSupervision::Read().

75  {
76  ExpectToken(is, binary, "<DiscriminativeSupervision>");
77  ExpectToken(is, binary, "<Weight>");
78  ReadBasicType(is, binary, &weight);
79  ExpectToken(is, binary, "<NumSequences>");
80  ReadBasicType(is, binary, &num_sequences);
81  ExpectToken(is, binary, "<FramesPerSeq>");
82  ReadBasicType(is, binary, &frames_per_sequence);
83  KALDI_ASSERT(frames_per_sequence > 0 &&
84  num_sequences > 0);
85 
86  ExpectToken(is, binary, "<NumAli>");
87  ReadIntegerVector(is, binary, &num_ali);
88 
89  ExpectToken(is, binary, "<DenLat>");
90  {
91  Lattice *lat = NULL;
92  if (!ReadLattice(is, binary, &lat) || lat == NULL) {
93  // We can't return error status from this function so we
94  // throw an exception.
95  KALDI_ERR << "Error reading Lattice from stream";
96  }
97  den_lat = *lat;
98  delete lat;
99  TopSort(&den_lat);
100  }
101 
102  ExpectToken(is, binary, "</DiscriminativeSupervision>");
103 }
kaldi::ReadLattice
bool ReadLattice(std::istream &is, bool binary, Lattice **lat)
Definition: kaldi-lattice.cc:417
kaldi::ReadBasicType
void ReadBasicType(std::istream &is, bool binary, T *t)
ReadBasicType is the name of the read function for bool, integer types, and floating-point types...
Definition: io-funcs-inl.h:55
kaldi::ReadIntegerVector
void ReadIntegerVector(std::istream &is, bool binary, std::vector< T > *v)
Function for reading STL vector of integer types.
Definition: io-funcs-inl.h:232
kaldi::ExpectToken
void ExpectToken(std::istream &is, bool binary, const char *token)
ExpectToken tries to read in the given token, and throws an exception on failure. ...
Definition: io-funcs.cc:191
kaldi::Lattice
fst::VectorFst< LatticeArc > Lattice
Definition: kaldi-lattice.h:44
KALDI_ERR
#define KALDI_ERR
Definition: kaldi-error.h:147
KALDI_ASSERT
#define KALDI_ASSERT(cond)
Definition: kaldi-error.h:185

◆ Swap()

void Swap ( DiscriminativeSupervision *  other)

Definition at line 34 of file discriminative-supervision.cc.

References DiscriminativeSupervision::den_lat, DiscriminativeSupervision::frames_per_sequence, DiscriminativeSupervision::num_ali, DiscriminativeSupervision::num_sequences, kaldi::swap(), and DiscriminativeSupervision::weight.

Referenced by kaldi::nnet3::MergeSupervision(), and NnetDiscriminativeSupervision::Swap().

34  {
35  std::swap(weight, other->weight);
36  std::swap(num_sequences, other->num_sequences);
37  std::swap(frames_per_sequence, other->frames_per_sequence);
38  std::swap(num_ali, other->num_ali);
39  std::swap(den_lat, other->den_lat);
40 }
kaldi::swap
void swap(basic_filebuf< CharT, Traits > &x, basic_filebuf< CharT, Traits > &y)
Definition: basic-filebuf.h:275

◆ Write()

void Write ( std::ostream &  os,
bool  binary 
) const

Definition at line 51 of file discriminative-supervision.cc.

References DiscriminativeSupervision::den_lat, DiscriminativeSupervision::frames_per_sequence, KALDI_ASSERT, KALDI_ERR, DiscriminativeSupervision::num_ali, DiscriminativeSupervision::num_sequences, DiscriminativeSupervision::weight, kaldi::WriteBasicType(), kaldi::WriteIntegerVector(), kaldi::WriteLattice(), and kaldi::WriteToken().

Referenced by NnetDiscriminativeSupervision::Write().

51  {
52  WriteToken(os, binary, "<DiscriminativeSupervision>");
53  WriteToken(os, binary, "<Weight>");
54  WriteBasicType(os, binary, weight);
55  WriteToken(os, binary, "<NumSequences>");
56  WriteBasicType(os, binary, num_sequences);
57  WriteToken(os, binary, "<FramesPerSeq>");
58  WriteBasicType(os, binary, frames_per_sequence);
59  KALDI_ASSERT(frames_per_sequence > 0 &&
60  num_sequences > 0);
61 
62  WriteToken(os, binary, "<NumAli>");
63  WriteIntegerVector(os, binary, num_ali);
64 
65  WriteToken(os, binary, "<DenLat>");
66  if (!WriteLattice(os, binary, den_lat)) {
67  // We can't return error status from this function so we
68  // throw an exception.
69  KALDI_ERR << "Error writing denominator lattice to stream";
70  }
71 
72  WriteToken(os, binary, "</DiscriminativeSupervision>");
73 }
KALDI_ERR
#define KALDI_ERR
Definition: kaldi-error.h:147
kaldi::WriteToken
void WriteToken(std::ostream &os, bool binary, const char *token)
The WriteToken functions are for writing nonempty sequences of non-space characters.
Definition: io-funcs.cc:134
kaldi::WriteLattice
bool WriteLattice(std::ostream &os, bool binary, const Lattice &t)
Definition: kaldi-lattice.cc:388
KALDI_ASSERT
#define KALDI_ASSERT(cond)
Definition: kaldi-error.h:185
kaldi::WriteIntegerVector
void WriteIntegerVector(std::ostream &os, bool binary, const std::vector< T > &v)
Function for writing STL vectors of integer types.
Definition: io-funcs-inl.h:198
kaldi::WriteBasicType
void WriteBasicType(std::ostream &os, bool binary, T t)
WriteBasicType is the name of the write function for bool, integer types, and floating-point types...
Definition: io-funcs-inl.h:34

Member Data Documentation

◆ den_lat

Lattice den_lat

Definition at line 104 of file discriminative-supervision.h.

Referenced by DiscriminativeSupervision::Check(), DiscriminativeComputation::DiscriminativeComputation(), DiscriminativeSupervisionSplitter::DiscriminativeSupervisionSplitter(), DiscriminativeSupervisionSplitter::GetFrameRange(), DiscriminativeSupervision::Initialize(), kaldi::discriminative::MergeSupervision(), DiscriminativeSupervision::operator==(), DiscriminativeSupervision::Read(), DiscriminativeSupervision::Swap(), and DiscriminativeSupervision::Write().

◆ frames_per_sequence

int32 frames_per_sequence

Definition at line 91 of file discriminative-supervision.h.

Referenced by DiscriminativeSupervision::Check(), NnetDiscriminativeSupervision::CheckDim(), DiscriminativeComputation::Compute(), DiscriminativeSupervisionSplitter::DiscriminativeSupervisionSplitter(), DiscriminativeSupervisionSplitter::GetFrameRange(), DiscriminativeSupervision::Initialize(), DiscriminativeComputation::LookupNnetOutput(), kaldi::discriminative::MergeSupervision(), NnetDiscriminativeSupervision::NnetDiscriminativeSupervision(), DiscriminativeSupervision::operator==(), kaldi::nnet3::ProcessFile(), DiscriminativeComputation::ProcessPosteriors(), DiscriminativeSupervision::Read(), DiscriminativeSupervision::Swap(), and DiscriminativeSupervision::Write().

◆ num_ali

std::vector<int32> num_ali

Definition at line 97 of file discriminative-supervision.h.

Referenced by DiscriminativeSupervision::Check(), DiscriminativeComputation::Compute(), DiscriminativeComputation::ComputeObjfAndDeriv(), DiscriminativeSupervisionSplitter::GetFrameRange(), DiscriminativeSupervision::Initialize(), DiscriminativeComputation::LookupNnetOutput(), kaldi::discriminative::MergeSupervision(), DiscriminativeSupervision::operator==(), DiscriminativeSupervision::Read(), DiscriminativeSupervision::Swap(), and DiscriminativeSupervision::Write().

◆ num_sequences

int32 num_sequences

Definition at line 85 of file discriminative-supervision.h.

Referenced by DiscriminativeSupervision::Check(), NnetDiscriminativeSupervision::CheckDim(), DiscriminativeComputation::Compute(), DiscriminativeSupervisionSplitter::DiscriminativeSupervisionSplitter(), DiscriminativeSupervisionSplitter::GetFrameRange(), DiscriminativeSupervision::Initialize(), DiscriminativeComputation::LookupNnetOutput(), kaldi::discriminative::MergeSupervision(), NnetDiscriminativeSupervision::NnetDiscriminativeSupervision(), DiscriminativeSupervision::operator==(), kaldi::nnet3::ProcessFile(), DiscriminativeComputation::ProcessPosteriors(), DiscriminativeSupervision::Read(), DiscriminativeSupervision::Swap(), and DiscriminativeSupervision::Write().

◆ weight

BaseFloat weight

Definition at line 79 of file discriminative-supervision.h.

Referenced by DiscriminativeComputation::Compute(), DiscriminativeSupervisionSplitter::GetFrameRange(), DiscriminativeSupervision::Initialize(), kaldi::discriminative::MergeSupervision(), DiscriminativeSupervision::operator==(), DiscriminativeComputation::ProcessPosteriors(), DiscriminativeSupervision::Read(), DiscriminativeSupervision::Swap(), and DiscriminativeSupervision::Write().

The documentation for this struct was generated from the following files: