Contents

module Torch
  module Utils
    module Data
      class Subset < Dataset
        def initialize(dataset, indices)
          @dataset = dataset
          @indices = indices
        end

        def [](idx)
          @dataset[@indices[idx]]
        end

        def length
          @indices.length
        end
        alias_method :size, :length

        def to_a
          @indices.map { |i| @dataset[i] }
        end
      end
    end
  end
end

Version data entries

41 entries across 41 versions & 1 rubygems

Version	Path
torch-rb-0.3.2	lib/torch/utils/data/subset.rb