Class for passing data to Torch's nn.StochasticGradient

torch_dataset.lua

-- Use the class rock to create a Dataset class
-- that can be used by nn.StochasticGradient in Torch
local class = require 'class'

local Dataset = class('Dataset')

function Dataset:__init(inputs, labels)
    self.inputs = inputs
    self.labels = labels
end

function Dataset:size()
    return self.inputs:size()[1]
end

-- Try looking keys up in original __index before
-- treating the key as an index into self.data
local t = Dataset.__index
Dataset.__index = function (self, k)
    if t[k] == nil then
        return {self.inputs[k], self.labels[k]}
    else
        return t[k]
    end
end

return {Dataset = Dataset}