diff options
| -rw-r--r-- | decoder/aligner.cc | 22 |
1 files changed, 20 insertions, 2 deletions
diff --git a/decoder/aligner.cc b/decoder/aligner.cc index 3f0c7347..43d4e0ce 100644 --- a/decoder/aligner.cc +++ b/decoder/aligner.cc @@ -1,12 +1,14 @@ #include "aligner.h" +#include <cstdio> +#include <set> + #include "array2d.h" #include "hg.h" #include "sentence_metadata.h" #include "inside_outside.h" #include "viterbi.h" #include "alignment_pharaoh.h" -#include <set> using namespace std; @@ -152,6 +154,22 @@ struct TransitionEventWeightFunction { } }; +inline void WriteProbGrid(const Array2D<prob_t>& m, ostream* pos) { + ostream& os = *pos; + char b[1024]; + for (int i=0; i<m.width(); ++i) { + for (int j=0; j<m.height(); ++j) { + if (m(i,j) == prob_t::Zero()) { + os << "\t---X---"; + } else { + snprintf(b, 1024, "%0.5f", static_cast<double>(m(i,j))); + os << '\t' << b; + } + } + os << '\n'; + } +} + // this code is rather complicated since it must deal with generating alignments // when lattices are specified as input as well as with models that do not generate // full sentence pairs (like lexical alignment models) @@ -245,7 +263,7 @@ void AlignerTools::WriteAlignment(const Lattice& src_lattice, } if (out == &cout) { // TODO need to do some sort of verbose flag - cerr << align << endl; + WriteProbGrid(align, &cerr); cerr << grid << endl; } (*out) << TD::GetString(src_sent) << " ||| " << TD::GetString(trg_sent) << " ||| "; |