diff options
Diffstat (limited to 'lm/builder/interpolate.hh')
-rw-r--r-- | lm/builder/interpolate.hh | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/lm/builder/interpolate.hh b/lm/builder/interpolate.hh index 55a55428f..0acece926 100644 --- a/lm/builder/interpolate.hh +++ b/lm/builder/interpolate.hh @@ -18,7 +18,7 @@ class Interpolate { public: // Normally vocab_size is the unigram count-1 (since p(<s>) = 0) but might // be larger when the user specifies a consistent vocabulary size. - explicit Interpolate(uint64_t vocab_size, const util::stream::ChainPositions &backoffs, const std::vector<uint64_t> &prune_thresholds); + explicit Interpolate(uint64_t vocab_size, const util::stream::ChainPositions &backoffs, const std::vector<uint64_t> &prune_thresholds, bool output_q_); void Run(const util::stream::ChainPositions &positions); @@ -26,6 +26,7 @@ class Interpolate { float uniform_prob_; util::stream::ChainPositions backoffs_; const std::vector<uint64_t> prune_thresholds_; + bool output_q_; }; }} // namespaces |