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
commit816f33174b2c2938a3df9c75b2834da6f5184a3e (patch)
tree2577931ef5ebc3a94c9f75692e8274c1b8eb15f8 /klm/search/weights.hh
parentf794c881ccaf4aa0646300dea1e1f6e7a307e019 (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