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)
添加回答
舉報