README.md in tensor_stream-0.1.1 vs README.md in tensor_stream-0.1.2

- old
+ new

@@ -55,12 +55,12 @@ 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 # Mean squared error @@ -69,11 +69,11 @@ optimizer = TensorStream::Train::GradientDescentOptimizer.new(learning_rate).minimize(cost) # 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| train_X.zip(train_Y).each do |x,y| sess.run(optimizer, feed_dict: {X => x, Y => y}) @@ -103,9 +103,10 @@ ## Issues - This is an early preview release and many things still don't work - Performance is not great, at least until the opencl and/or sciruby backends are complete +- However if you really need an op supported please feel free to file a pull request with the corresponding failing test (see spec/operation_spec.rb) ## Development After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.