From 367167680e156ac611c5f1db9f9ff7e66d51a8fe Mon Sep 17 00:00:00 2001 From: nunzip Date: Wed, 27 Feb 2019 22:49:16 +0000 Subject: Reshape labels after predict --- cgan.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/cgan.py b/cgan.py index ad3d194..fa05311 100644 --- a/cgan.py +++ b/cgan.py @@ -207,13 +207,14 @@ class CGAN(): labels_test[i*1000:] = i labels_val[i*500:] = i + train_data = self.generator.predict([noise_train, labels_train]) + test_data = self.generator.predict([noise_test, labels_test]) + val_data = self.generator.predict([noise_val, labels_val]) + labels_train = keras.utils.to_categorical(labels_train, 10) labels_test = keras.utils.to_categorical(labels_test, 10) labels_val = keras.utils.to_categorical(labels_val, 10) - train_data = self.generator.predict([noise_train, labels_train]) - test_data = self.generator.predict([noise_test, labels_test]) - val_data = self.generator.predict([noise_val, labels_val]) return train_data, test_data, val_data, labels_train, labels_test, labels_val -- cgit v1.2.3-54-g00ecf