Commit ·
d35a8da
1
Parent(s): 5ffae0a
Update README.md
Browse files
README.md
CHANGED
|
@@ -18,10 +18,15 @@
|
|
| 18 |
Функция потерь:
|
| 19 |
|
| 20 |
def vae_loss(x, y):
|
|
|
|
| 21 |
x = K.reshape(x, shape=(batch_size, 28*28))
|
|
|
|
| 22 |
y = K.reshape(y, shape=(batch_size, 28*28))
|
|
|
|
| 23 |
loss = K.sum(K.square(x-y), axis=-1)
|
|
|
|
| 24 |
kl_loss = -0.5 * K.sum(1 + z_log_var - K.square(z_mean) - K.exp(z_log_var), axis=-1)
|
|
|
|
| 25 |
return loss + kl_loss
|
| 26 |
|
| 27 |
Размеры обучающего, валидационного и тестового наборов данных, соответственно:
|
|
|
|
| 18 |
Функция потерь:
|
| 19 |
|
| 20 |
def vae_loss(x, y):
|
| 21 |
+
|
| 22 |
x = K.reshape(x, shape=(batch_size, 28*28))
|
| 23 |
+
|
| 24 |
y = K.reshape(y, shape=(batch_size, 28*28))
|
| 25 |
+
|
| 26 |
loss = K.sum(K.square(x-y), axis=-1)
|
| 27 |
+
|
| 28 |
kl_loss = -0.5 * K.sum(1 + z_log_var - K.square(z_mean) - K.exp(z_log_var), axis=-1)
|
| 29 |
+
|
| 30 |
return loss + kl_loss
|
| 31 |
|
| 32 |
Размеры обучающего, валидационного и тестового наборов данных, соответственно:
|