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.