Contents

require "dnn"

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

x = Numo::SFloat[[0, 0], [1, 0], [0, 1], [1, 1]]
y = Numo::SFloat[[0], [1], [1], [0]]

model = Model.new

model << InputLayer.new(2)
model << Dense.new(16)
model << ReLU.new
model << Dense.new(1)
model << SigmoidWithLoss.new

model.compile(SGD.new)

model.train(x, y, 20000, batch_size: 4, verbose: false)

p model.predict(x)

Version data entries

11 entries across 11 versions & 1 rubygems

Version	Path
ruby-dnn-0.6.7	examples/xor_example.rb
ruby-dnn-0.6.6	examples/xor_example.rb
ruby-dnn-0.6.5	examples/xor_example.rb
ruby-dnn-0.6.4	examples/xor_example.rb
ruby-dnn-0.6.3	examples/xor_example.rb
ruby-dnn-0.6.2	examples/xor_example.rb
ruby-dnn-0.6.1	examples/xor_example.rb
ruby-dnn-0.6.0	examples/xor_example.rb
ruby-dnn-0.5.12	examples/xor_example.rb
ruby-dnn-0.5.11	examples/xor_example.rb
ruby-dnn-0.5.10	examples/xor_example.rb