Skip to content

Commit 56335c5

Browse files
xionghhcszsdonghao
authored andcommitted
fix the bug of tensorlayer.utils.predict (#426)
1 parent c1491b3 commit 56335c5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorlayer/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -325,7 +325,7 @@ def predict(sess, network, X, x, y_op, batch_size=None):
325325
if result is None:
326326
result = result_a
327327
else:
328-
result = np.vstack((result, result_a)) # TODO: https://github.com/tensorlayer/tensorlayer/issues/288
328+
result = np.hstack((result, result_a)) # TODO: https://github.com/tensorlayer/tensorlayer/issues/288
329329
if result is None:
330330
if len(X) % batch_size != 0:
331331
dp_dict = dict_to_one(network.all_drop)
@@ -343,7 +343,7 @@ def predict(sess, network, X, x, y_op, batch_size=None):
343343
}
344344
feed_dict.update(dp_dict)
345345
result_a = sess.run(y_op, feed_dict=feed_dict)
346-
result = np.vstack((result, result_a)) # TODO: https://github.com/tensorlayer/tensorlayer/issues/288
346+
result = np.hstack((result, result_a)) # TODO: https://github.com/tensorlayer/tensorlayer/issues/288
347347
return result
348348

349349

0 commit comments

Comments
 (0)