keras中的本地连接1D自动编码器

2024-10-06 10:34:27 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图在Keras中创建一个LocallyConnected1D自动编码器,方法是从this tutorial重新使用一个“尽可能简单”的密集自动编码器。你知道吗

我不断得到下面的错误,我相信这是由我的input_shape维度造成的。你知道吗

Traceback (most recent call last):
File "localdendritic.py", line 38, in <module>
kernel_size=6)
File "localdendritic.py", line 15, in __init__
activation='relu')(input_placeholder)
File "/Users/me/anaconda3/lib/python3.6/site-packages/keras/engine/topology.py", line 573, in __call__
self.assert_input_compatibility(inputs)
File "/Users/me/anaconda3/lib/python3.6/site-packages/keras/engine/topology.py", line 472, in assert_input_compatibility
str(K.ndim(x)))
ValueError: Input 0 is incompatible with layer encoded_layer: expected ndim=3, found ndim=2

我的代码在下面。我试过把input_shape数组改成[None, 1, input_size][1, 1, input_size][1, input_size][None, input_size],但似乎什么都没变。我想我缺少了一些关于输入形状的见解。你知道吗

import numpy as np
from keras.models import Model, Sequential
from keras.layers import Input, LocallyConnected1D


class Localautoencoder:

  def __init__(self, input_size, encoded_size, kernel_size, **kwargs):

    input_shape = np.array([input_size])
    input_placeholder = Input(shape=(input_size, 1))
    encoded = LocallyConnected1D(encoded_size, kernel_size,
                                 input_shape=input_shape,
                                 name='encoded_layer',
                                 activation='relu')(input_placeholder)
    decoded = LocallyConnected1D(input_size, kernel_size,
                                activation='sigmoid',
                                name='decoded_layer')(encoded)
    self.localae = Model(input_placeholder, decoded)
    self.encoder = Model(input_placeholder, encoded)
    encoded_input = Input(shape=(1, encoded_size))
    decoded_layer = self.localae.layers[-1]
    self.decoder = Model(encoded_input, decoded_layer(encoded_input))
    self.localae.compile(optimizer='adam', loss='binary_crossentropy')


from keras.datasets import mnist
(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train = x_train.astype('float32')/255.
x_test = x_test.astype('float32')/255.
x_train = x_train.reshape((len(x_train), np.prod(x_train.shape[1:])))
x_test = x_test.reshape((len(x_test), np.prod(x_test.shape[1:])))
print(x_train.shape)
print(x_test.shape)

lae = Localautoencoder(input_size=x_train.shape[1],
                   encoded_size=100,
                   kernel_size=6)

Tags: pytestselflayerinputsizetrainkernel
1条回答
网友
1楼 · 发布于 2024-10-06 10:34:27

一个LocallyConnected1D层接受一个三维输入,但是输入的占位符只有二维。解决这个问题的方法是添加一个Reshape层,将2D输入转换成3D输入。你知道吗

相关问题 更多 >