README.md in ruby-dnn-1.1.0 vs README.md in ruby-dnn-1.1.1

- old
+ new

@@ -2,11 +2,11 @@ [![Gem Version](https://badge.fury.io/rb/ruby-dnn.svg)](https://badge.fury.io/rb/ruby-dnn) [![Build Status](https://travis-ci.org/unagiootoro/ruby-dnn.svg?branch=master)](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}" ```