README.md in ruby-dnn-1.1.0 vs README.md in ruby-dnn-1.1.1
- old
+ new
@@ -2,11 +2,11 @@
[](https://badge.fury.io/rb/ruby-dnn)
[](https://travis-ci.org/unagiootoro/ruby-dnn)
ruby-dnn is a ruby deep learning library. This library supports full connected neural network and convolution neural network
and recurrent neural network.
-Currently, you can get 99% accuracy with MNIST and 78% with CIFAR 10.
+Currently, you can get 99% accuracy with MNIST and 82% with CIFAR 10.
## Installation
Add this line to your application's Gemfile:
@@ -40,10 +40,9 @@
model << Dense.new(10)
model.setup(Adam.new, SoftmaxCrossEntropy.new)
model.train(x_train, y_train, 10, batch_size: 128, test: [x_test, y_test])
-
accuracy, loss = model.evaluate(x_test, y_test)
puts "accuracy: #{accuracy}"
puts "loss: #{loss}"
```