diff options
| author |   Keisuke Kuroyanagi <ksk@google.com> | 2014-09-09 18:33:55 +0900 |
|---|---|---|
| committer | Keisuke Kuroyanagi <ksk@google.com> | 2014-09-10 19:51:48 +0900 |
| commit | 2111e3abc9c9c0ea0350b8470532bf636b78cdd7 (patch) | |
| tree | e7917dd5af7648c15ef477a8f642960dc395beb3 /native/jni/src/suggest/core/dicnode | |
| parent | 11a48f92a517b2c8b1899bc57ab9dfe6c48209eb (diff) | |
| download | latinime-2111e3abc9c9c0ea0350b8470532bf636b78cdd7.tar.gz latinime-2111e3abc9c9c0ea0350b8470532bf636b78cdd7.tar.xz latinime-2111e3abc9c9c0ea0350b8470532bf636b78cdd7.zip | |
Introduce WordAttributes to get word probability and flags.
Bug: 14425059
Change-Id: Iee11d038e0893d7ddd6c52447907f8c55fecb6a5
Diffstat (limited to 'native/jni/src/suggest/core/dicnode')
| -rw-r--r-- | native/jni/src/suggest/core/dicnode/dic_node_utils.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/native/jni/src/suggest/core/dicnode/dic_node_utils.cpp b/native/jni/src/suggest/core/dicnode/dic_node_utils.cpp index 19f92cc0b..26c7e3357 100644 --- a/native/jni/src/suggest/core/dicnode/dic_node_utils.cpp +++ b/native/jni/src/suggest/core/dicnode/dic_node_utils.cpp @@ -72,10 +72,10 @@ namespace latinime { if (dicNode->hasMultipleWords() && !dicNode->isValidMultipleWordSuggestion()) { return static_cast<float>(MAX_VALUE_FOR_WEIGHTING); } - const int probability = dictionaryStructurePolicy->getProbabilityOfWordInContext( + const WordAttributes wordAttributes = dictionaryStructurePolicy->getWordAttributesInContext( dicNode->getPrevWordIds(), dicNode->getWordId(), multiBigramMap); // TODO: This equation to calculate the improbability looks unreasonable. Investigate this. - const float cost = static_cast<float>(MAX_PROBABILITY - probability) + const float cost = static_cast<float>(MAX_PROBABILITY - wordAttributes.getProbability()) / static_cast<float>(MAX_PROBABILITY); return cost; } |