Class: Torch::Utils::Data::DataLoader

Inherits:
Object
  • Object
show all
Includes:
Enumerable
Defined in:
lib/torch/utils/data/data_loader.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(dataset, batch_size: 1) ⇒ DataLoader

Returns a new instance of DataLoader.



9
10
11
12
 
# File 'lib/torch/utils/data/data_loader.rb', line 9

def initialize(dataset, batch_size: 1)
  @dataset = dataset
  @batch_size = batch_size
end

Instance Attribute Details

#dataset ⇒ Object (readonly)

Returns the value of attribute dataset.



7
8
9
 
# File 'lib/torch/utils/data/data_loader.rb', line 7

def dataset
  @dataset
end

Instance Method Details

#each ⇒ Object 



14
15
16
17
18
19
 
# File 'lib/torch/utils/data/data_loader.rb', line 14

def each
  size.times do |i|
    start_index = i * @batch_size
    yield @dataset[start_index...(start_index + @batch_size)]
  end
end

#size ⇒ Object 



21
22
23
 
# File 'lib/torch/utils/data/data_loader.rb', line 21

def size
  (@dataset.size / @batch_size.to_f).ceil
end