From bed05c401cd70c7a7bda24308afa35eab15f239e Mon Sep 17 00:00:00 2001 From: Kai Xiong <980265291@qq.com> Date: Thu, 26 Jul 2018 09:41:45 +0800 Subject: [PATCH] fix a bug in line 88 ----> del extra index '[0]' --- datanway.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/datanway.py b/datanway.py index 55a4c39..189cf63 100644 --- a/datanway.py +++ b/datanway.py @@ -85,6 +85,6 @@ def packslice(self, data_pack, numsamples): support_cacheX.append(slice_x) support_cacheY.append(keras.utils.to_categorical(slice_y,self.classes_per_set)) - target_cacheY.append(keras.utils.to_categorical(target_y,self.classes_per_set)[0]) + target_cacheY.append(keras.utils.to_categorical(target_y,self.classes_per_set)) - return np.array(support_cacheX), np.array(support_cacheY), np.array(target_cacheY) \ No newline at end of file + return np.array(support_cacheX), np.array(support_cacheY), np.array(target_cacheY)