diff options
| -rw-r--r-- | native/jni/src/suggest/policyimpl/dictionary/utils/forgetting_curve_utils.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/native/jni/src/suggest/policyimpl/dictionary/utils/forgetting_curve_utils.cpp b/native/jni/src/suggest/policyimpl/dictionary/utils/forgetting_curve_utils.cpp index bac4d4eba..c7d3df984 100644 --- a/native/jni/src/suggest/policyimpl/dictionary/utils/forgetting_curve_utils.cpp +++ b/native/jni/src/suggest/policyimpl/dictionary/utils/forgetting_curve_utils.cpp @@ -72,7 +72,8 @@ const ForgettingCurveUtils::ProbabilityTable ForgettingCurveUtils::sProbabilityT const int elapsedTimeStepCount = getElapsedTimeStepCount(historicalInfo->getTimeStamp(), headerPolicy->getForgettingCurveDurationToLevelDown()); return sProbabilityTable.getProbability( - headerPolicy->getForgettingCurveProbabilityValuesTableId(), historicalInfo->getLevel(), + headerPolicy->getForgettingCurveProbabilityValuesTableId(), + std::min(std::max(historicalInfo->getLevel(), 0), MAX_LEVEL), std::min(std::max(elapsedTimeStepCount, 0), MAX_ELAPSED_TIME_STEP_COUNT)); } |