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)); }