Merge pull request #2687 from QiJune/fix_input_shape

fix input shape of train_y in get started doc
8 years ago · ea641da51b
parent 45a78a4f20 9bb33f27f8
commit ea641da51b
1 changed files with 1 additions and 1 deletions
--- a/doc/getstarted/concepts/src/train.py
+++ b/doc/getstarted/concepts/src/train.py
@ -31,7 +31,7 @@ def event_handler(event):
 # define training dataset reader
 def train_reader():
    train_x = np.array([[1, 1], [1, 2], [3, 4], [5, 2]])
-    train_y = np.array([-2, -3, -7, -7])
+    train_y = np.array([[-2], [-3], [-7], [-7]])
    def reader():
        for i in xrange(train_y.shape[0]):