diff options
author | Kenneth Heafield <github@kheafield.com> | 2012-09-13 11:15:32 +0100 |
---|---|---|
committer | Kenneth Heafield <github@kheafield.com> | 2012-09-13 11:15:32 +0100 |
commit | 816f33174b2c2938a3df9c75b2834da6f5184a3e (patch) | |
tree | 2577931ef5ebc3a94c9f75692e8274c1b8eb15f8 /klm/search/weights.hh | |
parent | f794c881ccaf4aa0646300dea1e1f6e7a307e019 (diff) |
It compiles.
Diffstat (limited to 'klm/search/weights.hh')
-rw-r--r-- | klm/search/weights.hh | 17 |
1 files changed, 10 insertions, 7 deletions
diff --git a/klm/search/weights.hh b/klm/search/weights.hh index 4a4388c7..df1c419f 100644 --- a/klm/search/weights.hh +++ b/klm/search/weights.hh @@ -23,25 +23,28 @@ class Weights { // Parses weights, sets lm_weight_, removes it from map_. explicit Weights(StringPiece text); - search::Score DotNoLM(StringPiece text) const; + // Just the three scores we care about adding. + Weights(Score lm, Score oov, Score word_penalty); - search::Score LM() const { return lm_; } + Score DotNoLM(StringPiece text) const; - search::Score OOV() const { return oov_; } + Score LM() const { return lm_; } - search::Score WordPenalty() const { return word_penalty_; } + Score OOV() const { return oov_; } + + Score WordPenalty() const { return word_penalty_; } // Mostly for testing. - const boost::unordered_map<std::string, search::Score> &GetMap() const { return map_; } + const boost::unordered_map<std::string, Score> &GetMap() const { return map_; } private: float Steal(const std::string &str); - typedef boost::unordered_map<std::string, search::Score> Map; + typedef boost::unordered_map<std::string, Score> Map; Map map_; - search::Score lm_, oov_, word_penalty_; + Score lm_, oov_, word_penalty_; }; } // namespace search |