mirror of
https://gitlab.futo.org/keyboard/latinime.git
synced 2024-09-28 14:54:30 +01:00
Quit using dicNode.getUnigramProbability().
Bug: 14425059 Change-Id: I192070cc11e5d46c8413ebc19982d6a8c93577fc
This commit is contained in:
parent
521e2382da
commit
c32356c229
@ -54,15 +54,18 @@ namespace latinime {
|
||||
current.swap(next);
|
||||
}
|
||||
|
||||
int maxUnigramProbability = NOT_A_PROBABILITY;
|
||||
int maxProbability = NOT_A_PROBABILITY;
|
||||
for (const DicNode &dicNode : current) {
|
||||
if (!dicNode.isTerminalDicNode()) {
|
||||
continue;
|
||||
}
|
||||
const WordAttributes wordAttributes =
|
||||
dictionaryStructurePolicy->getWordAttributesInContext(dicNode.getPrevWordIds(),
|
||||
dicNode.getWordId(), nullptr /* multiBigramMap */);
|
||||
// dicNode can contain case errors, accent errors, intentional omissions or digraphs.
|
||||
maxUnigramProbability = std::max(maxUnigramProbability, dicNode.getUnigramProbability());
|
||||
maxProbability = std::max(maxProbability, wordAttributes.getProbability());
|
||||
}
|
||||
return maxUnigramProbability;
|
||||
return maxProbability;
|
||||
}
|
||||
|
||||
/* static */ void DictionaryUtils::processChildDicNodes(
|
||||
|
@ -48,7 +48,8 @@ class Traversal {
|
||||
virtual int getTerminalCacheSize() const = 0;
|
||||
virtual bool isPossibleOmissionChildNode(const DicTraverseSession *const traverseSession,
|
||||
const DicNode *const parentDicNode, const DicNode *const dicNode) const = 0;
|
||||
virtual bool isGoodToTraverseNextWord(const DicNode *const dicNode) const = 0;
|
||||
virtual bool isGoodToTraverseNextWord(const DicNode *const dicNode,
|
||||
const int probability) const = 0;
|
||||
|
||||
protected:
|
||||
Traversal() {}
|
||||
|
@ -21,6 +21,7 @@
|
||||
#include "suggest/core/dicnode/dic_node_vector.h"
|
||||
#include "suggest/core/dictionary/dictionary.h"
|
||||
#include "suggest/core/dictionary/digraph_utils.h"
|
||||
#include "suggest/core/dictionary/word_attributes.h"
|
||||
#include "suggest/core/layout/proximity_info.h"
|
||||
#include "suggest/core/policy/dictionary_structure_with_buffer_policy.h"
|
||||
#include "suggest/core/policy/traversal.h"
|
||||
@ -412,7 +413,11 @@ void Suggest::weightChildNode(DicTraverseSession *traverseSession, DicNode *dicN
|
||||
*/
|
||||
void Suggest::createNextWordDicNode(DicTraverseSession *traverseSession, DicNode *dicNode,
|
||||
const bool spaceSubstitution) const {
|
||||
if (!TRAVERSAL->isGoodToTraverseNextWord(dicNode)) {
|
||||
const WordAttributes wordAttributes =
|
||||
traverseSession->getDictionaryStructurePolicy()->getWordAttributesInContext(
|
||||
dicNode->getPrevWordIds(), dicNode->getWordId(),
|
||||
traverseSession->getMultiBigramMap());
|
||||
if (!TRAVERSAL->isGoodToTraverseNextWord(dicNode, wordAttributes.getProbability())) {
|
||||
return;
|
||||
}
|
||||
|
||||
|
@ -161,9 +161,8 @@ class TypingTraversal : public Traversal {
|
||||
return true;
|
||||
}
|
||||
|
||||
AK_FORCE_INLINE bool isGoodToTraverseNextWord(const DicNode *const dicNode) const {
|
||||
// TODO: Quit using unigram probability and use probability in the context.
|
||||
const int probability = dicNode->getUnigramProbability();
|
||||
AK_FORCE_INLINE bool isGoodToTraverseNextWord(const DicNode *const dicNode,
|
||||
const int probability) const {
|
||||
if (probability < ScoringParams::THRESHOLD_NEXT_WORD_PROBABILITY) {
|
||||
return false;
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user