Class: DNN::Callbacks::EarlyStopping

Inherits:
Callback
  • Object
show all
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 collapse

Constructor Details

#initialize(trigger, tolerance) ⇒ EarlyStopping

Returns a new instance of EarlyStopping.



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

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

Instance Method Details

#after_epoch ⇒ Object 



68
69
70
 
# File 'lib/dnn/core/callbacks.rb', line 68

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

#after_train_on_batch ⇒ Object 



64
65
66
 
# File 'lib/dnn/core/callbacks.rb', line 64

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