Merge "Fix a bug where the returned number of bigrams was incorrect"
This commit is contained in:
commit
2a7224a611
1 changed files with 3 additions and 2 deletions
|
@ -137,9 +137,10 @@ int BigramDictionary::getBigrams(unsigned short *prevWord, int prevWordLength, i
|
|||
// codesSize == 0 means we are trying to find bigram predictions.
|
||||
if (codesSize < 1 || checkFirstCharacter(bigramBuffer)) {
|
||||
const int frequency = UnigramDictionary::MASK_ATTRIBUTE_FREQUENCY & bigramFlags;
|
||||
addWordBigram(bigramBuffer, length, frequency);
|
||||
if (addWordBigram(bigramBuffer, length, frequency)) {
|
||||
++bigramCount;
|
||||
}
|
||||
}
|
||||
++bigramCount;
|
||||
} while (0 != (UnigramDictionary::FLAG_ATTRIBUTE_HAS_NEXT & bigramFlags));
|
||||
return bigramCount;
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue