人工神经网络的训练开始时,存在尺寸过大的问题

时间:2019-12-22 10:31:23

标签: python tensorflow

我的数据集和代码都没有问题。我现在只是更改了一些参数,例如“批处理大小”。我还将所有内容都上传到了Google Colab。

现在我总是收到此消息:

  

输入   /tensorflow-2.1.0/python3.6/tensorflow_core/python/keras/engine/training_v2.py:677 map_fn           batch_size =无)       /tensorflow-2.1.0/python3.6/tensorflow_core/python/keras/engine/training.py:2410   _standardize_tensors           exception_prefix ='输入')       /tensorflow-2.1.0/python3.6/tensorflow_core/python/keras/engine/training_utils.py:573   standardize_input_data           'with shape'+ str(data_shape))

     

ValueError:检查输入时出错:预期lstm_2_input具有3   尺寸,但具有形状的数组(无,1)

有人可以解释一下我,为什么会出现这个问题,我该怎么办? PS。我不是专家,也许可以从代码中给您更多。

simple_lstm_model = tf.keras.models.Sequential([
tf.keras.layers.LSTM(8, input_shape=x_train_uni.shape[-2:]),
tf.keras.layers.Dense(1)])                                  
simple_lstm_model.compile(optimizer='adam', loss='mse')

0 个答案:

没有答案