【发布时间】:2019-01-02 06:12:49
【问题描述】:
我正在尝试使用 MNIST 示例数据集在 R 中实现一个简单的 keras autoencoder。我从博客中得到了我的例子,但它不起作用。我几乎得到了0 % 的准确度。
目标是将每个28 x 28 图像(784 个条目)压缩成一个包含 32 个条目的向量:
这是我的代码:
library(keras)
mnist <- dataset_mnist()
x_train <- mnist$train$x
# reshape
x_train <- array_reshape(x_train, c(nrow(x_train), 784))
x_train <- x_train / 255
model <- keras_model_sequential()
model %>%
layer_dense(units = 32, activation = 'relu', input_shape = c(784)) %>%
layer_dense(units=784, activation='sigmoid')
model %>% compile(
loss = 'categorical_crossentropy',
optimizer = 'adam',
metrics = c('accuracy')
)
history <- model %>% fit(
x_train, x_train,
epochs = 15, batch_size = 128,
validation_split = 0.2
)
【问题讨论】:
标签: r keras compression mnist autoencoder