diff options
| -rw-r--r-- | cgan.py | 7 |
1 files changed, 4 insertions, 3 deletions
@@ -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 |