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 | 8505fdfdf0bc4ce9acec42e1980a2fdd4f254109 (patch) | |
tree | 20f941c7f2a06a73337d5c6fb24ab1da49888399 /klm/search/weights.hh | |
parent | 58da794845e192cdae3b5ba96cacf234d6a66ed0 (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 |