Sha256: 239e041076c298c4fec418262ac3e2a0f0c693c3615b001d97c8f9e7d95de34d

Contents?: true

Size: 1.06 KB

Versions: 14

Compression:

Stored size: 1.06 KB

Contents

require "dnn"
require "dnn/lib/mnist"
#require "numo/linalg/autoloader"

include DNN::Layers
include DNN::Activations
include DNN::Optimizers
Model = DNN::Model
MNIST = DNN::MNIST

x_train, y_train = MNIST.load_train
x_test, y_test = MNIST.load_test

x_train = Numo::SFloat.cast(x_train).reshape(x_train.shape[0], 28, 28, 1)
x_test = Numo::SFloat.cast(x_test).reshape(x_test.shape[0], 28, 28, 1)

x_train /= 255
x_test /= 255

y_train = DNN::Util.to_categorical(y_train, 10)
y_test = DNN::Util.to_categorical(y_test, 10)

model = Model.new

model << InputLayer.new([28, 28, 1])

model << Conv2D.new(16, 5)
model << BatchNormalization.new
model << ReLU.new

model << MaxPool2D.new(2)

model << Conv2D.new(32, 5)
model << BatchNormalization.new
model << ReLU.new

model << Flatten.new

model << Dense.new(256)
model << BatchNormalization.new
model << ReLU.new
model << Dropout.new(0.5)

model << Dense.new(10)
model << SoftmaxWithLoss.new

model.compile(Adam.new)

model.train(x_train, y_train, 10, batch_size: 100, test: [x_test, y_test])

Version data entries

14 entries across 14 versions & 1 rubygems

Version Path
ruby-dnn-0.8.6 examples/mnist_conv2d_example.rb
ruby-dnn-0.8.5 examples/mnist_conv2d_example.rb
ruby-dnn-0.8.4 examples/mnist_conv2d_example.rb
ruby-dnn-0.8.3 examples/mnist_conv2d_example.rb
ruby-dnn-0.8.2 examples/mnist_conv2d_example.rb
ruby-dnn-0.8.1 examples/mnist_conv2d_example.rb
ruby-dnn-0.8.0 examples/mnist_conv2d_example.rb
ruby-dnn-0.7.3 examples/mnist_conv2d_example.rb
ruby-dnn-0.7.2 examples/mnist_conv2d_example.rb
ruby-dnn-0.7.1 examples/mnist_conv2d_example.rb
ruby-dnn-0.7.0 examples/mnist_conv2d_example.rb
ruby-dnn-0.6.10 examples/mnist_conv2d_example.rb
ruby-dnn-0.6.9 examples/mnist_conv2d_example.rb
ruby-dnn-0.6.8 examples/mnist_conv2d_example.rb