ajax/HeurAMS
0
1
Fork 0
Code Pull Requests Releases 13 Activity

规范代码

Browse Source
This commit is contained in:
Wang Zhiyu
2025-08-06 06:11:22 +08:00
parent 2cf2cdb33f
commit dd74dddf00
7 changed files with 344 additions and 261 deletions
Download Patch File Download Diff File Expand all files Collapse all files

134
particles.py
View File

@@ -3,37 +3,36 @@ import toml
import time
import auxiliary as aux
import time
class Electron():
class Electron:
"""电子: 记忆分析元数据及算法"""
algorithm = "SM-2" # 暂时使用 SM-2 算法进行记忆拟合, 考虑 SM-15 替代
algorithm = "SM-2" # 暂时使用 SM-2 算法进行记忆拟合, 考虑 SM-15 替代
def __init__(self, content: str, metadata: dict):
self.content = content
self.metadata = metadata
if metadata == {}:
#print("NULL")
# print("NULL")
self._default_init()
def _default_init(self):
defaults = {
'efactor': 2.5, # 易度系数, 越大越简单, 最大为5
'real_rept': 0, # (实际)重复次数
'rept': 0, # (有效)重复次数
'interval': 0, # 最佳间隔
'last_date': 0, # 上一次复习的时间戳
'next_date': 0, # 将要复习的时间戳
'is_activated': 0, # 激活状态
'efactor': 2.5, # 易度系数, 越大越简单, 最大为5
'real_rept': 0, # (实际)重复次数
'rept': 0, # (有效)重复次数
'interval': 0, # 最佳间隔
'last_date': 0, # 上一次复习的时间戳
'next_date': 0, # 将要复习的时间戳
'is_activated': 0, # 激活状态
# *NOTE: 此处"时间戳"是以天为单位的整数, 即 UNIX 时间戳除以一天的秒数取整
'last_modify': time.time() # 最后修改时间戳(此处是UNIX时间戳)
'last_modify': time.time() # 最后修改时间戳(此处是UNIX时间戳)
}
self.metadata = defaults
def activate(self):
self.metadata['is_activated'] = 1
self.metadata['last_modify'] = time.time()
def modify(self, var: str, value):
if var in self.metadata:
self.metadata[var] = value
@@ -41,7 +40,7 @@ class Electron():
else:
print(f"警告: '{var}' 非已知元数据字段")
def revisor(self, quality: int = 5, is_new_activation: bool = False):
def revisor(self, quality: int = 5, is_new_activation: bool = False):
"""SM-2 算法迭代决策机制实现
根据 quality(0 ~ 5) 进行参数迭代最佳间隔
quality 由主程序评估
@@ -51,41 +50,47 @@ class Electron():
"""
if quality == -1:
return -1
self.metadata['efactor'] = self.metadata['efactor'] + (0.1 - (5 - quality) * (0.08 + (5 - quality) * 0.02))
self.metadata['efactor'] = self.metadata['efactor'] + (
0.1 - (5 - quality) * (0.08 + (5 - quality) * 0.02)
)
self.metadata['efactor'] = max(1.3, self.metadata['efactor'])
if quality < 3:
# 若保留率低于 3，重置重复次数
self.metadata['rept'] = 0
self.metadata['interval'] = 0 # 设为0，以便下面重新计算 I(1)
self.metadata['rept'] = 0
self.metadata['interval'] = 0 # 设为0，以便下面重新计算 I(1)
else:
self.metadata['rept'] += 1
self.metadata['real_rept'] += 1
if is_new_activation: # 初次激活
if is_new_activation: # 初次激活
self.metadata['rept'] = 0
self.metadata['efactor'] = 2.5
if self.metadata['rept'] == 0: # 刚被重置或初次激活后复习
self.metadata['interval'] = 1 # I(1)
if self.metadata['rept'] == 0: # 刚被重置或初次激活后复习
self.metadata['interval'] = 1 # I(1)
elif self.metadata['rept'] == 1:
self.metadata['interval'] = 6 # I(2) 经验公式
self.metadata['interval'] = 6 # I(2) 经验公式
else:
self.metadata['interval'] = round(self.metadata['interval'] * self.metadata['efactor'])
self.metadata['interval'] = round(
self.metadata['interval'] * self.metadata['efactor']
)
self.metadata['last_date'] = aux.get_daystamp()
self.metadata['next_date'] = aux.get_daystamp() + self.metadata['interval']
self.metadata['last_modify'] = time.time()
def __str__(self):
return (f"记忆单元预览 \n"
f"内容: '{self.content}' \n"
f"易度系数: {self.metadata['efactor']:.2f} \n"
f"已经重复的次数: {self.metadata['rept']} \n"
f"下次间隔: {self.metadata['interval']} 天 \n"
f"下次复习日期时间戳: {self.metadata['next_date']}")
return (
f"记忆单元预览 \n"
f"内容: '{self.content}' \n"
f"易度系数: {self.metadata['efactor']:.2f} \n"
f"已经重复的次数: {self.metadata['rept']} \n"
f"下次间隔: {self.metadata['interval']} 天 \n"
f"下次复习日期时间戳: {self.metadata['next_date']}"
)
def __eq__(self, other):
if self.content == other.content:
@@ -94,7 +99,7 @@ class Electron():
def __hash__(self):
return hash(self.content)
def __getitem__(self, key):
if key == "content":
return self.content
@@ -102,14 +107,10 @@ class Electron():
return self.metadata[key]
else:
raise KeyError(f"Key '{key}' not found in metadata.")
def __setitem__(self, key, value):
if key == "content":
raise AttributeError("content 应为只读")
# 可以在此处添加更复杂的验证逻辑，例如只允许修改预定义的 metadata 键
# 或者根据键进行类型检查等。
self.metadata[key] = value
self.metadata['last_modify'] = time.time()
@@ -118,14 +119,15 @@ class Electron():
def __len__(self):
return len(self.metadata)
@staticmethod
def placeholder():
return Electron("电子对象样例内容", {})
class Nucleon:
"""核子: 材料元数据"""
def __init__(self, content: str, data: dict):
self.metadata = data
self.content = content
@@ -151,7 +153,8 @@ class Nucleon:
def placeholder():
return Nucleon("核子对象样例内容", {})
class NucleonUnion():
class NucleonUnion:
"""
替代原有 NucleonFile 类, 支持复杂逻辑
@@ -166,6 +169,7 @@ class NucleonUnion():
Parameters:
path (Path): 包含核子数据的文件路径。
"""
def __init__(self, path):
self.path = path
self.name = path.name.replace(path.suffix, "")
@@ -189,8 +193,10 @@ class NucleonUnion():
tmp = {i.content: i.metadata for i in self.nucleons}
toml.dump(tmp, f)
class ElectronUnion():
"取代原有 ElectronFile 类, 以支持复杂逻辑"
class ElectronUnion:
"""取代原有 ElectronFile 类, 以支持复杂逻辑"""
def __init__(self, path):
self.path = path
self.name = path.name.replace(path.suffix, "")
@@ -201,19 +207,21 @@ class ElectronUnion():
lst.append(Electron(i, all[i]))
self.electrons = lst
self.electrons_dict = {i.content: i for i in lst}
def sync(self):
"""同步 electrons_dict 中新增对到 electrons 中"""
self.electrons = self.electrons_dict.values()
def save(self):
#print(1)
# print(1)
with open(self.path, 'w') as f:
tmp = {i.content: i.metadata for i in self.electrons}
#print(tmp)
# print(tmp)
toml.dump(tmp, f)
"""class AtomicFile():
"""
class AtomicFile():
def __init__(self, path, type_="unknown"):
self.path = path
self.type_ = type_
@@ -237,30 +245,42 @@ class ElectronUnion():
return ""
def get_len(self):
return len(self.datalist)
"""
class Atom():
class Atom:
@staticmethod
def placeholder():
return (Electron.placeholder(), Nucleon.placeholder(), {})
@staticmethod
def advanced_placeholder():
return (
Electron("两只黄鹤鸣翠柳", {}),
Nucleon("两只黄鹤鸣翠柳", {"note": [],
"translation": "臣子李密陈言：我因命运不好，小时候遭遇到了不幸",
"keyword_note": {"险衅":"凶险祸患（这里指命运不好）", "夙":"早时，这里指年幼的时候", "闵":"通'悯'，指可忧患的事", "凶":"不幸，指丧父"}}),
Nucleon(
"两只黄鹤鸣翠柳",
{
"note": [],
"translation": "臣子李密陈言：我因命运不好，小时候遭遇到了不幸",
"keyword_note": {
"险衅": "凶险祸患（这里指命运不好）",
"夙": "早时，这里指年幼的时候",
"闵": "通'悯'，指可忧患的事",
"凶": "不幸，指丧父"
}
}
),
{
"keydata":{
"keydata": {
"note": "笔记",
"keyword_note": "关键词翻译",
"translation": "语句翻译"},
"testdata":{
"translation": "语句翻译"
},
"testdata": {
"additional_inf": ["translation", "note", "keyword_note"],
"fill_blank_test": ["translation"],
"draw_card_test": ["keyword_note"]
},
"is_new_activation": 0
})
}
)