EarlyStopping¶
- class unike.utils.EarlyStopping(save_path: str, patience: int = 2, verbose: bool = True, delta: float = 0)[源代码]¶
如果验证得分(越大越好)在给定的耐心后没有改善,则提前停止训练。
- __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_path: str¶
模型保存目录
- verbose: bool¶
如果为 True,则为每个验证得分改进打印一条消息。默认值:True