【问题标题】:Simple keras autoencoder with MNIST sample data not working带有 MNIST 样本数据的简单 keras 自动编码器不起作用
【发布时间】: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


    【解决方案1】:

    您想在这里使用binary_crossentropy 作为损失函数。 categorical_crossentropy 适用于多类分类问题(只有一个输出为 1),binary_crossentropy 适用于多标签分类。

    【讨论】:

    • 谢谢,我让它运行了,但还有一个问题。我将输入数据从图像更改为具有正值和负值的随机矩阵。解码后的输出只有正值。知道为什么会这样吗?似乎自动编码器只想要图像(即具有正值的矩阵)
    • 一个sigmoid函数的输出总是在0到1之间。
    猜你喜欢
    • 1970-01-01
    • 2020-12-09
    • 2021-06-09
    • 1970-01-01
    • 2019-01-21
    • 2021-02-07
    • 2017-11-27
    • 2017-11-19
    • 2021-12-26
    相关资源
    最近更新 更多