早期停止与tflearn

Question

我很难找出如何实施早期停止与tflearn。据说它通过在model.fit（）函数中使用回调来工作，但我不太清楚它是如何完成的... 这是网站上的示例，但它仍然需要一个Monitor类，我无法工作：

class MonitorCallback(tflearn.callbacks.Callback): 
    def __init__(self, api): 
     self.my_monitor_api = api 

    def on_epoch_end(self, training_state): 
     self.my_monitor_api.send({ 
      accuracy: training_state.global_acc, 
      loss: training_state.global_loss, 
     }) 

monitorCallback = new MonitorCallback(api) 
model = ... 

model.fit(..., callbacks=monitorCallback) 

有没有人有一个例子或如何做到这一点的想法？欢呼声

Answer 1

您使用的是哪个版本的tflearn？大多数情况下，您需要下载回购以利用该功能。 Early commits for the feature are dated on Aug 17 2016，但最新版本（编写时为v2.2.0）的日期为2016年8月10日，不包括它。也许这就解释了这个问题......