From 13595c9d459c8208de413dde7c1b03830a10b864 Mon Sep 17 00:00:00 2001
From: huanghaicheng1024 <58104303+huanghaicheng1024@users.noreply.github.com>
Date: Wed, 17 Mar 2021 09:19:45 +0800
Subject: [PATCH] Update vae_keras.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

83行`y_test_`修改为`y_test`，可以正常运行了
---
 vae_keras.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vae_keras.py b/vae_keras.py
index ca1f4ef..a3328ec 100644
--- a/vae_keras.py
+++ b/vae_keras.py
@@ -80,7 +80,7 @@ def sampling(args):
 
 x_test_encoded = encoder.predict(x_test, batch_size=batch_size)
 plt.figure(figsize=(6, 6))
-plt.scatter(x_test_encoded[:, 0], x_test_encoded[:, 1], c=y_test_)
+plt.scatter(x_test_encoded[:, 0], x_test_encoded[:, 1], c=y_test)
 plt.colorbar()
 plt.show()