【发布时间】:2021-02-09 18:02:13
【问题描述】:
我试图了解如何使用 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 层的文档似乎表明它可以处理向量...
【问题讨论】:
-
CONV1D 也是如此:stackoverflow.com/q/36992855/10375049 你需要传递 3D 数据而不是 2D
-
这是否意味着我根本不应该使用 Conv1D?
标签: python numpy tensorflow keras