166 lines
5.0 KiB
Python
166 lines
5.0 KiB
Python
from typing import TypedDict
|
|
|
|
import heurams.services.timer as timer
|
|
from heurams.services.logger import get_logger
|
|
|
|
from .base import BaseAlgorithm
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
class SM2Algorithm(BaseAlgorithm):
|
|
"""SuperMemo-2 算法实现
|
|
|
|
经典间隔重复算法, 基于 1987 年 Piotr Wozniak 设计的 SM-2.
|
|
通过维护 efactor (难度因子) 来调整复习间隔.
|
|
|
|
Attributes:
|
|
algo_name: "SM-2"
|
|
desc: SuperMemo2 (1987) 简单间隔重复调度器
|
|
"""
|
|
|
|
algo_name = "SM-2"
|
|
desc = "SuperMemo2 (1987) 简单间隔重复调度器"
|
|
|
|
class AlgodataDict(TypedDict):
|
|
efactor: float
|
|
real_rept: int
|
|
rept: int
|
|
interval: int
|
|
last_date: int
|
|
next_date: int
|
|
is_activated: int
|
|
last_modify: float
|
|
|
|
defaults = {
|
|
"efactor": 2.5,
|
|
"real_rept": 0,
|
|
"rept": 0,
|
|
"interval": 0,
|
|
"last_date": 0,
|
|
"next_date": 0,
|
|
"is_activated": 0,
|
|
"last_modify": timer.get_timestamp(),
|
|
}
|
|
|
|
@classmethod
|
|
def revisor(
|
|
cls, algodata: dict, feedback: int = 5, is_new_activation: bool = False
|
|
):
|
|
"""SM-2 算法迭代决策机制实现
|
|
|
|
根据 feedback (0-5) 更新 efactor 并计算下次复习间隔.
|
|
|
|
Args:
|
|
algodata: 算法数据字典
|
|
feedback: 记忆保留率量化参数 (0-5), -1 表示跳过
|
|
is_new_activation: 是否为首次激活
|
|
"""
|
|
logger.debug(
|
|
"SM2.revisor 开始, feedback: %d, is_new_activation: %s",
|
|
feedback,
|
|
is_new_activation,
|
|
)
|
|
|
|
if feedback == -1:
|
|
logger.debug("feedback 为 -1, 跳过更新")
|
|
return
|
|
|
|
algodata[cls.algo_name]["efactor"] = algodata[cls.algo_name]["efactor"] + (
|
|
0.1 - (5 - feedback) * (0.08 + (5 - feedback) * 0.02)
|
|
)
|
|
algodata[cls.algo_name]["efactor"] = max(
|
|
1.3, algodata[cls.algo_name]["efactor"]
|
|
)
|
|
logger.debug("更新 efactor: %f", algodata[cls.algo_name]["efactor"])
|
|
|
|
if feedback < 3:
|
|
algodata[cls.algo_name]["rept"] = 0
|
|
algodata[cls.algo_name]["interval"] = 0
|
|
logger.debug("feedback < 3, 重置 rept 和 interval")
|
|
else:
|
|
algodata[cls.algo_name]["rept"] += 1
|
|
logger.debug("递增 rept: %d", algodata[cls.algo_name]["rept"])
|
|
|
|
algodata[cls.algo_name]["real_rept"] += 1
|
|
logger.debug("递增 real_rept: %d", algodata[cls.algo_name]["real_rept"])
|
|
|
|
if is_new_activation:
|
|
algodata[cls.algo_name]["rept"] = 0
|
|
algodata[cls.algo_name]["efactor"] = 2.5
|
|
logger.debug("新激活, 重置 rept 和 efactor")
|
|
|
|
if algodata[cls.algo_name]["rept"] == 0:
|
|
algodata[cls.algo_name]["interval"] = 1
|
|
logger.debug("rept=0, 设置 interval=1")
|
|
elif algodata[cls.algo_name]["rept"] == 1:
|
|
algodata[cls.algo_name]["interval"] = 6
|
|
logger.debug("rept=1, 设置 interval=6")
|
|
else:
|
|
algodata[cls.algo_name]["interval"] = round(
|
|
algodata[cls.algo_name]["interval"] * algodata[cls.algo_name]["efactor"]
|
|
)
|
|
logger.debug(
|
|
"rept>1, 计算 interval: %d", algodata[cls.algo_name]["interval"]
|
|
)
|
|
|
|
algodata[cls.algo_name]["last_date"] = timer.get_daystamp()
|
|
algodata[cls.algo_name]["next_date"] = (
|
|
timer.get_daystamp() + algodata[cls.algo_name]["interval"]
|
|
)
|
|
algodata[cls.algo_name]["last_modify"] = timer.get_timestamp()
|
|
|
|
logger.debug(
|
|
"更新日期: last_date=%d, next_date=%d, last_modify=%f",
|
|
algodata[cls.algo_name]["last_date"],
|
|
algodata[cls.algo_name]["next_date"],
|
|
algodata[cls.algo_name]["last_modify"],
|
|
)
|
|
|
|
@classmethod
|
|
def is_due(cls, algodata):
|
|
"""判断是否应该复习
|
|
|
|
Args:
|
|
algodata: 算法数据字典
|
|
|
|
Returns:
|
|
True 表示到期, False 表示未到期
|
|
"""
|
|
result = algodata[cls.algo_name]["next_date"] <= timer.get_daystamp()
|
|
logger.debug(
|
|
"SM2.is_due: next_date=%d, current_daystamp=%d, result=%s",
|
|
algodata[cls.algo_name]["next_date"],
|
|
timer.get_daystamp(),
|
|
result,
|
|
)
|
|
return result
|
|
|
|
@classmethod
|
|
def get_rating(cls, algodata):
|
|
"""获取当前 efactor 作为评分信息
|
|
|
|
Args:
|
|
algodata: 算法数据字典
|
|
|
|
Returns:
|
|
efactor 值的字符串表示
|
|
"""
|
|
efactor = algodata[cls.algo_name]["efactor"]
|
|
logger.debug("SM2.rate: efactor=%f", efactor)
|
|
return str(efactor)
|
|
|
|
@classmethod
|
|
def nextdate(cls, algodata) -> int:
|
|
"""获取下一次复习日期
|
|
|
|
Args:
|
|
algodata: 算法数据字典
|
|
|
|
Returns:
|
|
下次复习的天数戳
|
|
"""
|
|
next_date = algodata[cls.algo_name]["next_date"]
|
|
logger.debug("SM2.nextdate: %d", next_date)
|
|
return next_date
|