gpt4 book ai didi

python - Keras 输入数据格式

转载 作者:行者123 更新时间:2023-11-30 08:49:19 24 4
gpt4 key购买 nike

我试图将自己的数据结构传递到 Keras 中的 LSTM 层,但模型的训练在早期就失败了。

我的目的是创建一个固定长度的序列并接收相同大小的输出序列。目前,顺序如下:

[0, 0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3]

这是一个列表,然后我使用 np.array() 将其转换为 np 数组。标签序列的构建方式类似。我总共有 8000 个这样的序列,所有序列的长度都是 12。

在我的 Keras 模型中,LSTM 是第一层。这要求我将数据的输入形状传递给模型。我尝试过填充几种形状,但没有成功。为了完整起见,这些是我的模型的前两行:

model = Sequential()
model.add(LSTM(100, input_shape=(8000, 12)))
model.add(Dense(nb_classes))
model.add(Activation('softmax'))

在训练期间我遇到了一些麻烦,我遇到了this邮政。这看起来与我想做的事情非常相似。我已经尝试过提供的解决方案,但不知何故配件仍然崩溃。我收到以下错误:

ValueError: Error when checking input: expected lstm_1_input to have 3 dimensions, but got array with shape (8000, 12)

有什么想法吗?

更新:对于 Parthosarathi Mukherjee,我添加了网络的其余部分。

最佳答案

model.add(LSTM(100, input_shape(8000, 12)))

改为使用

model.add(LSTM(100, input_shape(12,1)))

拟合之前,确保输入数据的形状为8000 x 12 x 1,而不是8000 x 12

关于python - Keras 输入数据格式,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44720548/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com