Sha256: c86e65788d77e2e3bea1805b22a6348a70c1eb0a854bcbb3024249e3db502d3d
Contents?: true
Size: 955 Bytes
Versions: 11
Compression:
Stored size: 955 Bytes
Contents
module DNN
# This module provides utility functions.
module Util
# Create a mini batch for "batch_size".
def self.get_minibatch(x, y, batch_size)
indexes = (0...x.shape[0]).to_a.sample(batch_size)
[x[indexes, false], y[indexes, false]]
end
# Categorize labels into "num_classes" classes.
def self.to_categorical(y, num_classes, narray_type = nil)
narray_type ||= y.class
y2 = narray_type.zeros(y.shape[0], num_classes)
y.shape[0].times do |i|
y2[i, y[i]] = 1
end
y2
end
# Perform numerical differentiation.
def self.numerical_grad(x, func)
(func.(x + 1e-7) - func.(x)) / 1e-7
end
# Convert hash to an object.
def self.load_hash(hash)
dnn_class = DNN.const_get(hash[:class])
if dnn_class.respond_to?(:load_hash)
return dnn_class.load_hash(hash)
end
dnn_class.new
end
end
end
Version data entries
11 entries across 11 versions & 1 rubygems