Module: DataModeler::Dataset::IteratingBasedOnNext

Included in:
DataModeler::Dataset, DataModeler::DatasetGen
Defined in:
lib/data_modeler/dataset/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/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/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/dataset_helper.rb', line 51

def to_a
  each.to_a
end