lib/dnn/core/models.rb in ruby-dnn-1.1.5 vs lib/dnn/core/models.rb in ruby-dnn-1.1.6

@@ -472,9 +472,18 @@
       def add_callback(callback)
         callback.model = self
         @callbacks << callback
       end
 
+      # Add lambda callback.
+      # @param [Symbol] event Event to execute callback.
+      # @yield Register the contents of the callback.
+      def add_lambda_callback(event, &block)
+        callback = Callbacks::LambdaCallback.new(event, &block)
+        callback.model = self
+        @callbacks << callback
+      end
+
       # Clear the callback function registered for each event.
       def clear_callbacks
         @callbacks = []
       end