#include <nnet-compute-discriminative.h>

Collaboration diagram for NnetDiscriminativeStats:

Public Member Functions
	NnetDiscriminativeStats ()

void	Print (std::string criterion)

void	Add (const NnetDiscriminativeStats &other)

Public Attributes
double	tot_t

double	tot_t_weighted

double	tot_num_count

double	tot_num_objf

double	tot_den_objf

Detailed Description

Definition at line 70 of file nnet-compute-discriminative.h.

Constructor & Destructor Documentation

◆ NnetDiscriminativeStats()

NnetDiscriminativeStats ( )

inline

Definition at line 80 of file nnet-compute-discriminative.h.

References NnetDiscriminativeUpdateOptions::criterion, kaldi::nnet2::NnetDiscriminativeUpdate(), and fst::Print().

80 { std::memset(this, 0, sizeof(*this)); }

Member Function Documentation

◆ Add()

void Add ( const NnetDiscriminativeStats & other )

Definition at line 376 of file nnet-compute-discriminative.cc.

References NnetDiscriminativeStats::tot_den_objf, NnetDiscriminativeStats::tot_num_count, NnetDiscriminativeStats::tot_num_objf, NnetDiscriminativeStats::tot_t, and NnetDiscriminativeStats::tot_t_weighted.

                                                                       {
   tot_t += other.tot_t;
   tot_t_weighted += other.tot_t_weighted;
   tot_num_count += other.tot_num_count;
   tot_num_objf += other.tot_num_objf;
   tot_den_objf += other.tot_den_objf;
 }

◆ Print()

void Print ( std::string criterion )

Definition at line 384 of file nnet-compute-discriminative.cc.

References KALDI_ASSERT, and KALDI_LOG.

Referenced by main(), and kaldi::nnet2::NnetDiscriminativeUpdateParallel().

                                                        {
   KALDI_ASSERT(criterion == "mmi" || criterion == "smbr" ||
                criterion == "mpfe");
 
   double avg_post_per_frame = tot_num_count / tot_t_weighted;
   KALDI_LOG << "Number of frames is " << tot_t
             << " (weighted: " << tot_t_weighted
             << "), average (num or den) posterior per frame is "
             << avg_post_per_frame;
   
   if (criterion == "mmi") {
     double num_objf = tot_num_objf / tot_t_weighted,
         den_objf = tot_den_objf / tot_t_weighted,
         objf = num_objf - den_objf;
     KALDI_LOG << "MMI objective function is " << num_objf << " - "
               << den_objf << " = " << objf << " per frame, over "
               << tot_t_weighted << " frames.";
   } else if (criterion == "mpfe") {
     double objf = tot_den_objf / tot_t_weighted; // this contains the actual
                                                  // summed objf
     KALDI_LOG << "MPFE objective function is " << objf
               << " per frame, over " << tot_t_weighted << " frames.";
   } else {
     double objf = tot_den_objf / tot_t_weighted; // this contains the actual
                                                  // summed objf
     KALDI_LOG << "SMBR objective function is " << objf
               << " per frame, over " << tot_t_weighted << " frames.";
   }
 }