优化日志
This commit is contained in:
@@ -542,7 +542,7 @@ class ExpressionLearner:
|
||||
# 获取 StyleLearner 实例
|
||||
learner = style_learner_manager.get_learner(chat_id)
|
||||
|
||||
logger.info(f"开始训练 StyleLearner: chat_id={chat_id}, 样本数={len(expr_list)}")
|
||||
logger.debug(f"开始训练 StyleLearner: chat_id={chat_id}, 样本数={len(expr_list)}")
|
||||
|
||||
# 为每个学习到的表达方式训练模型
|
||||
# 使用 situation 作为输入,style 作为目标
|
||||
@@ -566,12 +566,12 @@ class ExpressionLearner:
|
||||
|
||||
# 保存模型
|
||||
if learner.save(style_learner_manager.model_save_path):
|
||||
logger.info(f"StyleLearner 模型保存成功: {chat_id}")
|
||||
logger.debug(f"StyleLearner 模型保存成功: {chat_id}")
|
||||
else:
|
||||
logger.error(f"StyleLearner 模型保存失败: {chat_id}")
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"训练 StyleLearner 失败: {e}", exc_info=True)
|
||||
logger.error(f"训练 StyleLearner 失败: {e}")
|
||||
|
||||
return learnt_expressions
|
||||
return None
|
||||
|
||||
@@ -38,7 +38,7 @@ class ExpressorModel:
|
||||
self._candidates: dict[str, str] = {} # cid -> text (style)
|
||||
self._situations: dict[str, str] = {} # cid -> situation (不参与计算)
|
||||
|
||||
logger.info(
|
||||
logger.debug(
|
||||
f"ExpressorModel初始化完成 (alpha={alpha}, beta={beta}, gamma={gamma}, vocab_size={vocab_size}, use_jieba={use_jieba})"
|
||||
)
|
||||
|
||||
@@ -199,8 +199,6 @@ class ExpressorModel:
|
||||
with open(path, "wb") as f:
|
||||
pickle.dump(data, f)
|
||||
|
||||
logger.info(f"模型已保存到 {path}")
|
||||
|
||||
def load(self, path: str):
|
||||
"""
|
||||
从文件加载模型
|
||||
@@ -230,7 +228,7 @@ class ExpressorModel:
|
||||
for cid, tc in data["nb_token_counts"].items():
|
||||
self.nb.token_counts[cid] = defaultdict(float, tc)
|
||||
|
||||
logger.info(f"模型已从 {path} 加载")
|
||||
logger.debug(f"模型已从 {path} 加载")
|
||||
|
||||
def get_stats(self) -> dict:
|
||||
"""获取模型统计信息"""
|
||||
|
||||
@@ -24,7 +24,7 @@ class Tokenizer:
|
||||
import rjieba # noqa: F401
|
||||
|
||||
# rjieba 会自动初始化,无需手动调用
|
||||
logger.info("RJieba分词器初始化成功")
|
||||
logger.debug("RJieba分词器初始化成功")
|
||||
except ImportError:
|
||||
logger.warning("RJieba未安装,将使用字符级分词")
|
||||
self.use_jieba = False
|
||||
|
||||
@@ -51,8 +51,6 @@ class StyleLearner:
|
||||
"last_update": time.time(),
|
||||
}
|
||||
|
||||
logger.info(f"StyleLearner初始化成功: chat_id={chat_id}")
|
||||
|
||||
def add_style(self, style: str, situation: str | None = None) -> bool:
|
||||
"""
|
||||
动态添加一个新的风格
|
||||
@@ -178,7 +176,7 @@ class StyleLearner:
|
||||
logger.debug(f"被删除的风格样例(前5): {deleted_styles[:5]}")
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"清理风格失败: {e}", exc_info=True)
|
||||
logger.error(f"清理风格失败: {e}")
|
||||
|
||||
def learn_mapping(self, up_content: str, style: str) -> bool:
|
||||
"""
|
||||
@@ -271,7 +269,7 @@ class StyleLearner:
|
||||
return best_style, style_scores
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"预测style失败: {e}", exc_info=True)
|
||||
logger.error(f"预测style失败: {e}")
|
||||
return None, {}
|
||||
|
||||
def get_style_info(self, style: str) -> tuple[str | None, str | None]:
|
||||
@@ -373,7 +371,6 @@ class StyleLearner:
|
||||
with open(meta_path, "wb") as f:
|
||||
pickle.dump(meta_data, f)
|
||||
|
||||
logger.info(f"StyleLearner保存成功: {save_dir}")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
@@ -421,7 +418,6 @@ class StyleLearner:
|
||||
if "style_last_used" not in self.learning_stats:
|
||||
self.learning_stats["style_last_used"] = {}
|
||||
|
||||
logger.info(f"StyleLearner加载成功: {save_dir}")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
@@ -454,7 +450,7 @@ class StyleLearnerManager:
|
||||
# 确保保存目录存在
|
||||
os.makedirs(model_save_path, exist_ok=True)
|
||||
|
||||
logger.info(f"StyleLearnerManager初始化成功, 模型保存路径: {model_save_path}")
|
||||
logger.debug(f"StyleLearnerManager初始化成功, 模型保存路径: {model_save_path}")
|
||||
|
||||
def get_learner(self, chat_id: str, model_config: dict | None = None) -> StyleLearner:
|
||||
"""
|
||||
@@ -520,7 +516,7 @@ class StyleLearnerManager:
|
||||
if not learner.save(self.model_save_path):
|
||||
success = False
|
||||
|
||||
logger.info(f"保存所有StyleLearner {'成功' if success else '部分失败'}")
|
||||
logger.debug(f"保存所有StyleLearner {'成功' if success else '部分失败'}")
|
||||
return success
|
||||
|
||||
def cleanup_all_old_styles(self, ratio: float | None = None) -> dict[str, int]:
|
||||
@@ -540,7 +536,7 @@ class StyleLearnerManager:
|
||||
cleanup_results[chat_id] = cleaned
|
||||
|
||||
total_cleaned = sum(cleanup_results.values())
|
||||
logger.info(f"清理所有StyleLearner完成: 总共清理了 {total_cleaned} 个风格")
|
||||
logger.debug(f"清理所有StyleLearner完成: 总共清理了 {total_cleaned} 个风格")
|
||||
return cleanup_results
|
||||
|
||||
def apply_decay_all(self, factor: float | None = None):
|
||||
@@ -553,7 +549,7 @@ class StyleLearnerManager:
|
||||
for learner in self.learners.values():
|
||||
learner.apply_decay(factor)
|
||||
|
||||
logger.info("对所有StyleLearner应用知识衰减")
|
||||
logger.debug("对所有StyleLearner应用知识衰减")
|
||||
|
||||
def get_all_stats(self) -> dict[str, dict]:
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user