diff options
| author | nunzip <np.scarh@gmail.com> | 2019-02-27 22:43:40 +0000 |
|---|---|---|
| committer | nunzip <np.scarh@gmail.com> | 2019-02-27 22:43:40 +0000 |
| commit | 35216b30bdff05c04fd4846cfc8433b97218139f (patch) | |
| tree | 447e029c1083154d65a9591ca79c3db41aefa02f /cgan.py | |
| parent | 765a74824171f9c3d1ab80d561debf2053c3dc06 (diff) | |
| download | e4-gan-35216b30bdff05c04fd4846cfc8433b97218139f.tar.gz e4-gan-35216b30bdff05c04fd4846cfc8433b97218139f.tar.bz2 e4-gan-35216b30bdff05c04fd4846cfc8433b97218139f.zip | |
Reformat labels output
Diffstat (limited to 'cgan.py')
| -rw-r--r-- | cgan.py | 4 |
1 files changed, 4 insertions, 0 deletions
@@ -206,6 +206,10 @@ class CGAN(): labels_test[i*1000:] = i labels_val[i*500:] = i + 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]) |