Red Chainer : A deep learning framework
May 3, 2019 ยท View on GitHub
A flexible framework for neural network for Ruby
Description
It ported python's Chainer with Ruby.
Requirements
- Ruby 2.4 or later
Installation
Add this line to your application's Gemfile:
gem 'red-chainer'
And then execute:
$ bundle
Or install it yourself as:
$ gem install red-chainer
Usage
Run MNIST example
MNIST sample program is here
# when install Gemfile
$ bundle exec ruby examples/mnist/mnist.rb
# when install yourself
$ ruby examples/mnist/mnist.rb
Run MNIST example with GPU
On GPU machine, add gem 'cumo' on Gemfile.local and do bundle install.
Run the example with --gpu option whose value indicates GPU device ID such as:
$ bundle exec ruby examples/mnist/mnist.rb --gpu 0
Development
Run tests
$ bundle exec ruby test/run_test.rb
Run tests with Cumo
On GPU machine, add gem 'cumo' on Gemfile.local and do bundle install.
Run tests with RED_CHAINER_GPU environment variable whose value indicates GPU device ID such as:
$ bundle exec env RED_CHAINER_GPU=0 ruby test/run_test.rb
License
The MIT license. See LICENSE.txt for details.
Red Chainer implementation status
| Chainer 2.0 (Initial ported version) | Red Chainer (0.3.1) | example | |
|---|---|---|---|
| activation | 15 | 5 | LogSoftmax, ReLU, LeakyReLU, Sigmoid, Tanh |
| loss | 17 | 2 | SoftMax, MeanSquaredError |
| optimizer | 9 | 2 | Adam, MomentumSGDRule |
| connection | 12 | 2 | Linear, Convolution2D |
| pooling | 14 | 3 | Pooling2D, MaxPooling2D, AveragePooling2D |
| example | 31 | 3 | MNIST, Iris, CIFAR |
| GPU | use CuPy | use Cumo |