Contents

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

include Numo
include DNN::Layers
include DNN::Activations
include DNN::Optimizers
Model = DNN::Model
CIFAR10 = DNN::CIFAR10

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

x_train = SFloat.cast(x_train)
x_test = SFloat.cast(x_test)

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([32, 32, 3])

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

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

model << MaxPool2D.new(2)

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

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

model << MaxPool2D.new(2)

model << Conv2D.new(64, 5, padding: true)
model << BatchNormalization.new
model << ReLU.new

model << Conv2D.new(64, 5, padding: true)
model << BatchNormalization.new
model << ReLU.new

model << Flatten.new

model << Dense.new(512)
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

15 entries across 15 versions & 1 rubygems

Version	Path
ruby-dnn-0.5.9	examples/cifar10_example.rb
ruby-dnn-0.5.8	examples/cifar10_example.rb
ruby-dnn-0.5.7	examples/cifar10_example.rb
ruby-dnn-0.5.6	examples/cifar10_example.rb
ruby-dnn-0.5.5	examples/cifar10_example.rb
ruby-dnn-0.5.4	examples/cifar10_example.rb
ruby-dnn-0.5.3	examples/cifar10_example.rb
ruby-dnn-0.5.2	examples/cifar10_example.rb
ruby-dnn-0.5.1	examples/cifar10_example.rb
ruby-dnn-0.5.0	examples/cifar10_example.rb
ruby-dnn-0.4.4	examples/cifar10_example.rb
ruby-dnn-0.4.3	examples/cifar10_example.rb
ruby-dnn-0.4.2	examples/cifar10_example.rb
ruby-dnn-0.4.1	examples/cifar10_example.rb
ruby-dnn-0.4.0	examples/cifar10_example.rb