Sha256: 2e1f9b54be85598a0dd96045891ff7e721774048260b2e10ef127cc308b53c56

Contents?: true

Size: 504 Bytes

Versions: 7

Compression:

Stored size: 504 Bytes

Contents

require "dnn"

include DNN::Layers
include DNN::Activations
include DNN::Optimizers
include DNN::Losses
Model = DNN::Model
Utils = DNN::Utils

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.compile(SGD.new, SigmoidCrossEntropy.new)

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

p Utils.sigmoid(model.predict(x))

Version data entries

7 entries across 7 versions & 1 rubygems

Version Path
ruby-dnn-0.10.1 examples/xor_example.rb
ruby-dnn-0.10.0 examples/xor_example.rb
ruby-dnn-0.9.4 examples/xor_example.rb
ruby-dnn-0.9.3 examples/xor_example.rb
ruby-dnn-0.9.2 examples/xor_example.rb
ruby-dnn-0.9.1 examples/xor_example.rb
ruby-dnn-0.9.0 examples/xor_example.rb