diff options
Diffstat (limited to 'lm/builder/print.cc')
| -rw-r--r-- | lm/builder/print.cc | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/lm/builder/print.cc b/lm/builder/print.cc index 84bd81cad..c70e62ed6 100644 --- a/lm/builder/print.cc +++ b/lm/builder/print.cc @@ -42,14 +42,14 @@ PrintARPA::PrintARPA(const VocabReconstitute &vocab, const std::vector<uint64_t> util::WriteOrThrow(out_fd, as_string.data(), as_string.size()); } -void PrintARPA::Run(const ChainPositions &positions) { +void PrintARPA::Run(const util::stream::ChainPositions &positions) { util::scoped_fd closer(out_fd_); UTIL_TIMER("(%w s) Wrote ARPA file\n"); util::FakeOFStream out(out_fd_); for (unsigned order = 1; order <= positions.size(); ++order) { out << "\\" << order << "-grams:" << '\n'; for (NGramStream stream(positions[order - 1]); stream; ++stream) { - // Correcting for numerical precision issues. Take that IRST. + // Correcting for numerical precision issues. Take that IRST. out << std::min(0.0f, stream->Value().complete.prob) << '\t' << vocab_.Lookup(*stream->begin()); for (const WordIndex *i = stream->begin() + 1; i != stream->end(); ++i) { out << ' ' << vocab_.Lookup(*i); @@ -58,6 +58,7 @@ void PrintARPA::Run(const ChainPositions &positions) { if (backoff != 0.0) out << '\t' << backoff; out << '\n'; + } out << '\n'; } |