Fix: BoS prediction after inputting just once.
Change-Id: Ib69569ab6b6edfcc8c1d2c621b95de4127789ab6
This commit is contained in:
parent
2af3fc3275
commit
2cf5550749
1 changed files with 10 additions and 2 deletions
|
@ -146,8 +146,16 @@ void Ver4PatriciaTriePolicy::iterateNgramEntries(const WordIdArrayView prevWordI
|
||||||
if (!probabilityEntry.isValid()) {
|
if (!probabilityEntry.isValid()) {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
const int probability = probabilityEntry.hasHistoricalInfo() ?
|
int probability = NOT_A_PROBABILITY;
|
||||||
0 : probabilityEntry.getProbability();
|
if (probabilityEntry.hasHistoricalInfo()) {
|
||||||
|
// TODO: Quit checking count here.
|
||||||
|
// If count <= 1, the word can be an invaild word. The actual probability should
|
||||||
|
// be checked using getWordAttributesInContext() in onVisitEntry().
|
||||||
|
probability = probabilityEntry.getHistoricalInfo()->getCount() <= 1 ?
|
||||||
|
NOT_A_PROBABILITY : 0;
|
||||||
|
} else {
|
||||||
|
probability = probabilityEntry.getProbability();
|
||||||
|
}
|
||||||
listener->onVisitEntry(probability, entry.getWordId());
|
listener->onVisitEntry(probability, entry.getWordId());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue