Module: DataModeler::Dataset::IteratingBasedOnNext

Included in:
DataModeler::Dataset, DataModeler::DatasetGen
Defined in:
lib/data_modeler/dataset/helper.rb

Overview

Provides #each (which can return an Iterator) and #to_a based on #next

Instance Method Summary collapse

Instance Method Details

#each ⇒ nil|Iterator

Yields on each [inputs, targets] pair.

Returns:

  • (nil|Iterator) —

    ‘block_given? ? nil : Iterator`



32
33
34
35
36
37
 
# File 'lib/data_modeler/dataset/helper.rb', line 32

def each
  reset_iteration
  return enum_for(:each) unless block_given?
  loop { yield self.next }
  nil
end

#map ⇒ Array|Iterator Also known as: collect

Yields on each ‘[inputs, targets]` pair, collecting the input.

Returns:

  • (Array|Iterator) —

    ‘block_given? ? nil : Iterator`



41
42
43
44
45
 
# File 'lib/data_modeler/dataset/helper.rb', line 41

def map
  reset_iteration
  return enum_for(:collect) unless block_given?
  [].tap { |ret| loop { ret << yield(self.next) } }
end

#to_a ⇒ Array

Returns:

  • (Array) 


51
52
53
 
# File 'lib/data_modeler/dataset/helper.rb', line 51

def to_a
  each.to_a
end