早期停止與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日，不包括它。也許這就解釋了這個問題......