red-chainer

Red Chainer : A deep learning framework

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 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 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