summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorChris Dyer <cdyer@cs.cmu.edu>2011-03-08 15:33:37 -0500
committerChris Dyer <cdyer@cs.cmu.edu>2011-03-08 15:33:37 -0500
commitb8f64438e66e2335948492833c39dcc5afe72518 (patch)
treee4527728d6075272b95a48090879d795f3b7f70e
parent6001bfe4dcb637fe21fb7b47b4a375e9c28c33c5 (diff)
fix class-based lm
-rw-r--r--decoder/ff_klm.cc4
1 files changed, 2 insertions, 2 deletions
diff --git a/decoder/ff_klm.cc b/decoder/ff_klm.cc
index 446485f6..adc2c8bf 100644
--- a/decoder/ff_klm.cc
+++ b/decoder/ff_klm.cc
@@ -318,7 +318,7 @@ class KLanguageModelImpl {
vector<WordID> dummy;
int lc = 0;
cerr << " Loading word classes from " << file << " ...\n";
- AddWordToClassMapping_(TD::Convert("<s>"), TD::Convert("</s>"));
+ AddWordToClassMapping_(TD::Convert("<s>"), TD::Convert("<s>"));
AddWordToClassMapping_(TD::Convert("</s>"), TD::Convert("</s>"));
while(in) {
getline(in, line);
@@ -339,7 +339,7 @@ class KLanguageModelImpl {
word2class_map_.resize((word + 10) * 1.1, kCDEC_UNK);
assert(word2class_map_.size() > word);
}
- if(word2class_map_[word] == kCDEC_UNK) {
+ if(word2class_map_[word] != kCDEC_UNK) {
cerr << "Multiple classes for symbol " << TD::Convert(word) << endl;
abort();
}