summaryrefslogtreecommitdiff
path: root/klm/search/weights.hh
diff options
context:
space:
mode:
authorKenneth Heafield <github@kheafield.com>2012-09-13 11:15:32 +0100
committerKenneth Heafield <github@kheafield.com>2012-09-13 11:15:32 +0100
commit8505fdfdf0bc4ce9acec42e1980a2fdd4f254109 (patch)
tree20f941c7f2a06a73337d5c6fb24ab1da49888399 /klm/search/weights.hh
parent58da794845e192cdae3b5ba96cacf234d6a66ed0 (diff)
It compiles.
Diffstat (limited to 'klm/search/weights.hh')
-rw-r--r--klm/search/weights.hh17
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