使用 Conv1D 層時出現 ValueError

Question

我試圖了解如何使用 Conv1D 層從向量中提取特征。 這是我的代碼：

import tensorflow as tf
from tensorflow.keras import models, layers
import numpy as np

# make 100 40000-dimensional vectors:
x = []
for i in range(100):
  array_of_random_floats = np.random.random_sample((40000))
  x.append(array_of_random_floats)
x = np.asarray(x)

# make 100 80000-dimensional vectors:
y = []
for i in range(100):
  array_of_random_floats = np.random.random_sample((80000))
  y.append(array_of_random_floats)
y = np.asarray(y)


model = models.Sequential([
  layers.Input((40000,)),
  layers.Conv1D(padding='same', kernel_initializer='Orthogonal', filters=16, kernel_size=16, activation=None, strides=2),
  # ...
  layers.Dense(80000)
])

model.compile(loss='mse', optimizer='adam', metrics=['accuracy'])

history = model.fit(x=x,
                    y=y,
                    epochs=100)

這會產生以下錯誤：

ValueError: Input 0 of layer conv1d_20 is incompatible with the layer: : expected min_ndim=3, found ndim=2. Full shape received: (None, 40000)

我有點困惑，因為 Conv1D 層的文檔似乎表明它可以處理向量......

Answer 1

看來這只是您的尺寸錯誤。 我發現如果您在 Conv1D 圖層中指定輸入形狀並擴展x的尺寸，它會起作用。

model = models.Sequential([
  layers.Conv1D(..., input_shape=(None, 40000)),
  # ...
  layers.Dense(80000)
])

和

history = model.fit(x=np.expand_dims(x, 1), y=y, epochs=100)