Class: DNN::Callbacks::EarlyStopping

Inherits:

Callback

• Object
• Callback
• DNN::Callbacks::EarlyStopping

Defined in:

lib/dnn/core/callbacks.rb

Overview

A callback to stop training the model early after test on batch.

Instance Attribute Summary

Attributes inherited from Callback

#model

Instance Method Summary

• #after_epoch ⇒ Object
• #after_train_on_batch ⇒ Object
• #initialize(trigger, tolerance) ⇒ EarlyStopping constructor

  A new instance of EarlyStopping.

Constructor Details

#initialize(trigger, tolerance) ⇒ EarlyStopping

# File 'lib/dnn/core/callbacks.rb', line 50

def initialize(trigger, tolerance)
  @trigger = trigger
  @tolerance = tolerance
end

Instance Method Details

#after_epoch ⇒ Object

# File 'lib/dnn/core/callbacks.rb', line 59

def after_epoch
  throw :stop, "Early stopped." if judge_early_stopping_test
end

#after_train_on_batch ⇒ Object

# File 'lib/dnn/core/callbacks.rb', line 55

def after_train_on_batch
  throw :stop, "Early stopped." if judge_early_stopping_train
end