tensor_stream 0.1.1 → 0.1.2

data/lib/tensor_stream/tensor_shape.rb

@@ -1,7 +1,8 @@
 module TensorStream
+  # class that defines a shape for TensorFlow compatibility
   class TensorShape
     attr_accessor :rank, :shape
-    
+
     def initialize(shape, rank)
       @shape = shape
       @rank = rank
@@ -17,9 +18,9 @@ module TensorStream
     def [](index)
       @shape[index]
     end
-    
+
     def ndims
       shape.size
     end
   end
-end
+end

data/lib/tensor_stream/train/gradient_descent_optimizer.rb

@@ -4,14 +4,14 @@ module TensorStream
     class GradientDescentOptimizer
       attr_accessor :learning_rate
 
-      def initialize(learning_rate, options = {})
+      def initialize(learning_rate, _options = {})
         @learning_rate = learning_rate
       end
 
       def minimize(cost)
-        trainable_vars = TensorStream::Graph.get_default_graph.
-                                             get_collection(GraphKeys::GLOBAL_VARIABLES).
-                                             select(&:trainable)
+        trainable_vars = TensorStream::Graph.get_default_graph
+                                            .get_collection(GraphKeys::GLOBAL_VARIABLES)
+                                            .select(&:trainable)
 
         derivatives = TensorStream.gradients(cost, trainable_vars)
         trainable_vars.each_with_index.collect do |var, index|
@@ -20,4 +20,4 @@ module TensorStream
       end
     end
   end
-end
+end

data/lib/tensor_stream/train/saver.rb

@@ -2,16 +2,16 @@ require 'json'
 
 module TensorStream
   module Train
+    # High level class used for loading and saving variables
     class Saver
-      def save(session, outputfile,
-                global_step: nil,
-                latest_filename: nil,
-                meta_graph_suffix: 'meta',
-                write_meta_graph: true,
-                write_state: true,
-                strip_default_attrs: false)
+      def save(session, outputfile, global_step: nil,
+               latest_filename: nil,
+               meta_graph_suffix: 'meta',
+               write_meta_graph: true,
+               write_state: true,
+               strip_default_attrs: false)
         vars = TensorStream::Graph.get_default_graph.get_collection(GraphKeys::GLOBAL_VARIABLES)
-        
+
         variables = {}
         graph = {}
         gs =  eval_global_step(session, global_step)
@@ -34,12 +34,12 @@ module TensorStream
         path
       end
 
-      def restore(session, inputfile)
+      def restore(_session, inputfile)
         input_dump = JSON.parse(File.read(inputfile))
 
         vars = TensorStream::Graph.get_default_graph.get_collection(GraphKeys::GLOBAL_VARIABLES)
         vars.each do |variable|
-          variable.value = input_dump["variables"][variable.name]
+          variable.value = input_dump['variables'][variable.name]
         end
       end
 
@@ -48,9 +48,9 @@ module TensorStream
       def eval_global_step(session, global_step)
         return nil if global_step.nil?
 
-        if (global_step.is_a?(Tensor))
+        if global_step.is_a?(Tensor)
           session.last_session_context(global_step.name)
-        elsif (global_step.is_a?(String) || global_step.is_a?(Symbol))
+        elsif global_step.is_a?(String) || global_step.is_a?(Symbol)
           session.last_session_context(global_step)
         else
           global_step.to_i
@@ -58,4 +58,4 @@ module TensorStream
       end
     end
   end
-end
+end

data/lib/tensor_stream/trainer.rb

@@ -4,4 +4,4 @@ require 'tensor_stream/train/saver'
 module TensorStream
   module Train
   end
-end
+end

data/lib/tensor_stream/types.rb

@@ -1,6 +1,7 @@
 require 'ostruct'
 
 module TensorStream
+  # Convenience class for specifying valid data_types
   module Types
     def self.int16
       :int16
@@ -13,5 +14,17 @@ module TensorStream
     def self.int32
       :int32
     end
+
+    def self.float64
+      :float64
+    end
+
+    def self.string
+      :string
+    end
+
+    def self.boolean
+      :boolean
+    end
   end
-end
+end

data/lib/tensor_stream/variable.rb

@@ -1,17 +1,18 @@
 module TensorStream
+  # Class that defines a TensorStream variable
   class Variable < Tensor
     attr_accessor :trainable
     def initialize(data_type, rank, shape, options = {})
+      @graph = options[:graph] || TensorStream.get_default_graph
+
       @data_type = data_type
       @rank = rank
       @shape = TensorShape.new(shape, rank)
       @value = nil
-      @source = set_source(caller_locations)
-      @graph = options[:graph] || TensorStream.get_default_graph
+      @source = format_source(caller_locations)
+
       @name = options[:name] || build_name
-      if options[:initializer]
-        @initalizer_tensor = options[:initializer]
-      end
+      @initalizer_tensor = options[:initializer] if options[:initializer]
       @trainable = options.fetch(:trainable, true)
       @graph.add_variable(self, options)
     end
@@ -33,7 +34,7 @@ module TensorStream
       Operation.new(:assign_add, self, value)
     end
 
-    def to_math(tensor, name_only = false, max_depth = 99)
+    def to_math(_tensor, _name_only = false, _max_depth = 99)
       @name
     end
 
@@ -49,4 +50,4 @@ module TensorStream
       variables_initializer(TensorStream::GraphKeys::GLOBAL_VARIABLES)
     end
   end
-end
+end

data/lib/tensor_stream/version.rb

@@ -1,5 +1,5 @@
 module TensorStream
-  VERSION = '0.1.1'
+  VERSION = '0.1.2'.freeze
 
   def self.version
     VERSION

data/samples/iris.rb

@@ -68,15 +68,15 @@ y = tf.placeholder("float", shape: [nil, num_classes], name: 'y')
 
 # Store layers weight & bias
 weights = {
-    h1: tf.Variable(tf.random_normal([num_input, n_hidden_1]), name: 'h1'),
-    h2: tf.Variable(tf.random_normal([n_hidden_1, n_hidden_2]), name: 'h2'),
-    out: tf.Variable(tf.random_normal([n_hidden_2, num_classes]), name: 'out')
+    h1: tf.variable(tf.random_normal([num_input, n_hidden_1]), name: 'h1'),
+    h2: tf.variable(tf.random_normal([n_hidden_1, n_hidden_2]), name: 'h2'),
+    out: tf.variable(tf.random_normal([n_hidden_2, num_classes]), name: 'out')
 }
 
 biases = {
-    b1: tf.Variable(tf.random_normal([n_hidden_1]), name: 'b1'),
-    b2: tf.Variable(tf.random_normal([n_hidden_2]), name: 'b2'),
-    out: tf.Variable(tf.random_normal([num_classes]), name: 'b_out')
+    b1: tf.variable(tf.random_normal([n_hidden_1]), name: 'b1'),
+    b2: tf.variable(tf.random_normal([n_hidden_2]), name: 'b2'),
+    out: tf.variable(tf.random_normal([num_classes]), name: 'b_out')
 }
 
 
@@ -100,7 +100,7 @@ optimizer = TensorStream::Train::GradientDescentOptimizer.new(learning_rate).min
 # Initialize the variables (i.e. assign their default value)
 init = TensorStream.global_variables_initializer()
 
-TensorStream.Session do |sess|
+TensorStream.session do |sess|
   puts "init vars"
   sess.run(init)
   puts "Testing the untrained network..."

data/samples/linear_regression.rb

@@ -19,8 +19,8 @@ X = tf.placeholder("float")
 Y = tf.placeholder("float")
 
 # Set model weights
-W = tf.Variable(rand, name: "weight")
-b = tf.Variable(rand, name: "bias")
+W = tf.variable(rand, name: "weight")
+b = tf.variable(rand, name: "bias")
 
 # Construct a linear model
 pred = X * W + b
@@ -33,7 +33,7 @@ optimizer = TensorStream::Train::GradientDescentOptimizer.new(learning_rate).min
 # Initialize the variables (i.e. assign their default value)
 init = tf.global_variables_initializer()
 
-tf.Session do |sess|
+tf.session do |sess|
     start_time = Time.now
     sess.run(init)
     (0..training_epochs).each do |epoch|

data/samples/raw_neural_net_sample.rb

@@ -21,15 +21,15 @@ Y = tf.placeholder("float", shape: [nil, num_classes])
 
 # Store layers weight & bias
 @weights = {
-    h1: tf.Variable(tf.random_normal([num_input, n_hidden_1])),
-    h2: tf.Variable(tf.random_normal([n_hidden_1, n_hidden_2])),
-    out: tf.Variable(tf.random_normal([n_hidden_2, num_classes]))
+    h1: tf.variable(tf.random_normal([num_input, n_hidden_1])),
+    h2: tf.variable(tf.random_normal([n_hidden_1, n_hidden_2])),
+    out: tf.variable(tf.random_normal([n_hidden_2, num_classes]))
 }
 
 @biases = {
-    b1: tf.Variable(tf.random_normal([n_hidden_1])),
-    b2: tf.Variable(tf.random_normal([n_hidden_2])),
-    out: tf.Variable(tf.random_normal([num_classes]))
+    b1: tf.variable(tf.random_normal([n_hidden_1])),
+    b2: tf.variable(tf.random_normal([n_hidden_2])),
+    out: tf.variable(tf.random_normal([num_classes]))
 }
 
 

data/tensor_stream.gemspec

@@ -35,6 +35,7 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency "rspec", "~> 3.0"
   spec.add_development_dependency "awesome_print"
   spec.add_development_dependency "rubocop"
+  spec.add_development_dependency "pry-byebug"
   spec.add_dependency "deep_merge"
   spec.add_dependency "concurrent-ruby"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tensor_stream
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Joseph Emmanuel Dayo
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-05-11 00:00:00.000000000 Z
+date: 2018-05-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -80,6 +80,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry-byebug
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: deep_merge
   requirement: !ruby/object:Gem::Requirement
@@ -121,6 +135,7 @@ files:
 - ".gitignore"
 - ".rake_tasks~"
 - ".rspec"
+- ".rubocop.yml"
 - ".travis.yml"
 - CODE_OF_CONDUCT.md
 - Gemfile