From 5eb4cb6f773066572e241a27f39ed9c80a546f51 Mon Sep 17 00:00:00 2001 From: Minki Date: Sat, 2 Sep 2017 16:43:55 +0900 Subject: [PATCH] =?UTF-8?q?Update=20DataFeeder=20constructor=20=EB=B2=84?= =?UTF-8?q?=EA=B7=B8=20(#4)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Minki --- tensorflow/contrib/learn/python/learn/learn_io/data_feeder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/contrib/learn/python/learn/learn_io/data_feeder.py b/tensorflow/contrib/learn/python/learn/learn_io/data_feeder.py index 48d79ecbb..bafde464a 100644 --- a/tensorflow/contrib/learn/python/learn/learn_io/data_feeder.py +++ b/tensorflow/contrib/learn/python/learn/learn_io/data_feeder.py @@ -323,7 +323,7 @@ def __init__(self, self._x = dict([(k, check_array(v, v.dtype)) for k, v in list(x.items()) ]) if x_is_dict else check_array(x, x.dtype) self._y = None if y is None else \ - dict([(k, check_array(v, v.dtype)) for k, v in list(y.items())]) if x_is_dict else check_array(y, y.dtype) + dict([(k, check_array(v, v.dtype)) for k, v in list(y.items())]) if y_is_dict else check_array(y, y.dtype) # self.n_classes is not None means we're converting raw target indices to one-hot. if n_classes is not None: