Shortcuts

EarlyStopping

class pybind11_ke.utils.EarlyStopping(save_path: str, patience: int = 2, verbose: bool = True, delta: float = 0)[源代码]

如果验证得分(越大越好)在给定的耐心后没有改善,则提前停止训练。

__call__(score: float, model: Model)[源代码]

进行早停记录。

__init__(save_path: str, patience: int = 2, verbose: bool = True, delta: float = 0)[源代码]

创建 EarlyStopping 对象。

参数:
  • save_path (str) – 模型保存目录

  • patience (int) – 上次验证得分改善后等待多长时间。默认值:2

  • verbose (bool) – 如果为 True,则为每个验证得分改进打印一条消息。默认值:True

  • delta (float) – 监测数量的最小变化才符合改进条件。默认值:0

__weakref__

list of weak references to the object (if defined)

best_score: float

保存最好的得分

counter: int

计数变量

delta: float

监测数量的最小变化才符合改进条件。默认值:0

early_stop: bool

早停开关

patience: int

上次验证得分改善后等待多长时间。默认值:2

save_checkpoint(score: float, model: Model)[源代码]

当验证得分改善时保存模型。

save_path: str

模型保存目录

verbose: bool

如果为 True,则为每个验证得分改进打印一条消息。默认值:True

Docs

Access comprehensive developer documentation for Pybind11-OpenKE

View Docs