Method: MNIST.categorical

Defined in:
lib/nn/mnist.rb

.categorical(y_data) ⇒ Object



28
29
30
31
32
33
34
# File 'lib/nn/mnist.rb', line 28

def self.categorical(y_data)
  y_data = y_data.map do |label|
    classes = Array.new(10, 0)
    classes[label] = 1
    classes
  end
end