restore_best_weights发出keras提前停止

时间:2018-09-23 14:08:45

标签: keras keras-2

我在深度学习项目中使用了来自keras的早期停止服务。 here文档提到了恢复最佳权重的非常有用的想法。但是不知何故我还不能使用它。我在从蟒蛇安装的keras 2.2.2 / TF 1.10上。 调用很简单,如下所示。有什么问题吗?

es = EarlyStopping(监控='val_acc',min_delta = 1e-4,耐心=耐心_,详细= 1,恢复_最佳_权重=真实)

init ()获得了意外的关键字参数'restore_best_weights'

1 个答案:

答案 0 :(得分:12)

嗯,这是一个非常普遍的问题。网上的keras文档是从github的master分支生成的,因此那里仅提供了最近才引入的功能,但在任何Keras版本中都没有。

因此,restore_best_weights回调的EarlyStopping选项是在2018年8月18日添加的,而当前最新的Keras版本(2.2.2)是在2018年7月28日发布的。要使用Keras母版,等待Keras发行,或者只是不使用该选项。

更新:它现已包含在Keras 2.2.3版本中。