Keras阵列形状错误

时间:2018-06-28 08:20:50

标签: python tensorflow neural-network keras

当我传入的是一组形状 view?.pinTintColor = .blue 时,我如何得到神经网络的形状ValueError,但是我得到的错误是神经网络抱怨获取(8,1)

神经网络:

(1,)

错误:

>>> observation_dimension
(8,)
>>> q_network = Sequential([
    Dense(40, input_dim=observation_dimension, activation='relu'),
    Dense(40, activation='relu'),
    Dense(number_of_actions, activation='linear')
])
>>> obs
array([-0.00371828,  0.93953934, -0.37663383, -0.07161933,  0.00431531,
        0.08531308,  0.        ,  0.        ])
>>> obs.shape
(8,)

2 个答案:

答案 0 :(得分:2)

q_network.predict(obs.reshape(1, 8)) 提取了一批样本,如果您给它一个形状错误的样本,它将把第一个维度解释为批次。

一个简单的解决方案是添加一个值为一的维:

non-instantiatable

答案 1 :(得分:1)

predict方法需要一个二维数组,因此只需重塑obs

obs = np.reshape(obs,(-1,len(obs)))