rubyzero 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 998d9df9776d84a808d54a8e7472d4a9ddea9ee6d0c6732705d8204153016b4e
-  data.tar.gz: 77abc1c5ed65a055339150c4443a72622afd5cb8266c6feb17b707b73678e225
+  metadata.gz: 357575700d5bb1cb77bd1edf44327fa01c0837225a7c2035cd967484fb4c1379
+  data.tar.gz: 0010a684b0527243a7070b776bf85ceeb51ac1c85bffcfdb14783965dc851922
 SHA512:
-  metadata.gz: 1ecf6f49bb0d4a182f204f89fc89d456f51871a7707e4b15bd2d232580b9c62673b1e95ae6159ad3a08535a6d4440d916b6916e33065300ea3674026d75470d3
-  data.tar.gz: 5f5837d4f2ca18403ecafba1309ec8c2fe00d317da7dee746c2d5822c933a6d7817db739df0be988c66f1044113f50b2e1f09356210a9f8a1baf9be46bc7546d
+  metadata.gz: 1b3c14bae96029ed19cc992eb1e7b16f7529e7e0007bd511e11a917f3bfde962578140e8c48f16398f7c7cbac2657f608d6e6971e0b613a3a7b9a5de6964c5c7
+  data.tar.gz: 92cdda68e7333c9a6627dbc20c0945863a4bc7bb824fea1dd88702b4f20fc9a479542a601cd50ddd9bc84c45166863cee20e9633382338360f9006a7332593d8

data/README.md

@@ -1,8 +1,17 @@
 # Rubyzero
 
-Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/rubyzero`. To experiment with that code, run `bin/console` for an interactive prompt.
+A Simple deep learning library for Ruby.
 
-TODO: Delete this and the text above, and describe your gem
+## Example
+```ruby
+require 'rubyzero'
+include RubyZero
+model = L.mlp(2,5,1) # 多層パーセプトロンモデルを初期化(入力2次元, 隠れ層5次元, 出力層1次元)
+data = RubyZero::Data::Presets::Xor.new() # XOR演算のデータセット
+trainer = Utils::Trainer.new(model) # 学習を勝手にしてくれるやつ
+trainer.train(data, data, num_epochs:100) # 学習を自動的にするメソッド
+p model # ついでにモデルの中身を表示
+```
 
 ## Installation
 

data/lib/rubyzero/core/functions/activations.rb

@@ -1,7 +1,7 @@
 module RubyZero::Core::Functions
     class ReLU < Function
         def forward(x)
-            @path_through = RubyZero::Core::Tensor.new(x.data < 0)
+            @path_through = RubyZero::Core::Tensor.new(x.data > 0)
             @path_through.cast_to(RubyZero::FloatTensor)
             return x * @path_through
         end
@@ -16,5 +16,9 @@ module RubyZero::Core::Functions
             data = 1.0 / (1.0 + nmath.exp(-x.data))
             return RubyZero::Core::Tensor.new(data)
         end
+        def backward(dy)
+            x = @inputs[0]
+            return [ F.sigmoid(self.forward(x)) * (x.ones_like - F.sigmoid(self.forward(x))) * dy ]
+        end
     end
 end

data/lib/rubyzero/core/functions/function.rb

@@ -1,4 +1,5 @@
 require_relative '../exceptions.rb'
+require 'unicode_plot'
 
 module RubyZero::Core::Functions
     # Function class
@@ -25,5 +26,11 @@ module RubyZero::Core::Functions
         def inspect
             return "#<#{self.class}>"
         end
+        def self.plot(range: (-10..10).step(0.01))
+            inputs = range.to_a
+            outputs = inputs.map{|t| self.new().call(RubyZero::Core::Tensor.new(t)).data[0]}
+            plot = UnicodePlot.lineplot(inputs, outputs, name: self.name)
+            plot.render
+        end
     end
 end

data/lib/rubyzero/core/functions/operators.rb

@@ -68,20 +68,6 @@ module RubyZero::Core::Functions
         end
     end
 
-    class Log < Function
-        def forward(x1)
-            nmath = x1.device.xmo::NMath
-            new_arr = nmath.log(x1.data)
-            new_t = RubyZero::Core::Tensor.new(new_arr, device: x1.device)
-            return new_t
-        end
-
-        def backward(dy)
-            x1 = @inputs[0]
-            return [dy / x1]
-        end
-    end
-
     class MulScalar < Function
         def initialize(scalar)
             @scalar = scalar

data/lib/rubyzero/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Rubyzero
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/tests/test.rb

@@ -0,0 +1,6 @@
+require_relative '../lib/rubyzero'
+L = RubyZero::L
+F = RubyZero::F
+
+RubyZero::Core::Functions::ReLU.plot
+RubyZero::Core::Functions::Sigmoid.plot

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rubyzero
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - uthree
@@ -73,6 +73,7 @@ files:
 - lib/rubyzero/version.rb
 - note.txt
 - rubyzero.gemspec
+- tests/test.rb
 homepage: https://github.com/uthree/RubyZero
 licenses: []
 metadata: