1 回答
TA贡献1828条经验 获得超6个赞
这里有一种可能性......您最终需要简单地处理维度并进行适当的重塑
encoding_dim = 20
input_shape = (1,200,4)
n_sample = 100
X = np.random.uniform(0,1, (n_sample,) + input_shape)
autoencoder = Sequential()
autoencoder.add(Flatten(input_shape=input_shape))
autoencoder.add(Dense(200, activation='relu'))
autoencoder.add(Dense(100, activation='relu'))
autoencoder.add(Dense(80, activation='relu'))
autoencoder.add(Dense(encoding_dim, activation='relu'))
#decoder
autoencoder.add(Dense(80, activation='relu'))
autoencoder.add(Dense(100, activation='relu'))
autoencoder.add(Dense(200, activation='relu'))
autoencoder.add(Dense(np.prod(input_shape), activation='linear'))
autoencoder.add(Reshape(input_shape))
print(autoencoder.summary())
autoencoder.compile('adam', 'mse')
autoencoder.fit(X,X, epochs=3)
添加回答
举报