fix yT calculation

LKH-1 · Nov 13, 2016 · 2cf2fcd · 2cf2fcd
1 parent fdd1427
commit 2cf2fcd
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 32 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/src/predict.py b/src/predict.py
@@ -51,5 +51,6 @@ def predict(stock, data):
     with tf.Session() as sess:
         tf.train.Saver().restore(sess, '../models' + stock + '_model.ckpt')
         prediction = sess.run([yT], feed_dict={x: data})
+        prediction = prediction[0][:, 0]
         print(prediction)
         return prediction