当我传入的是一组形状 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,)
答案 0 :(得分:2)
q_network.predict(obs.reshape(1, 8))
提取了一批样本,如果您给它一个形状错误的样本,它将把第一个维度解释为批次。
一个简单的解决方案是添加一个值为一的维:
non-instantiatable
答案 1 :(得分:1)
predict
方法需要一个二维数组,因此只需重塑obs
:
obs = np.reshape(obs,(-1,len(obs)))