首页 > 解决方案 > ValueError:检查输入时出错:预期dense_13_input的形状为(3,),但数组的形状为(1,)

问题描述

我试图用我的 keras 模型预测手动编码的数据,但它不起作用:

print(np.array([5.1, 3.5, 1.4]).shape)
prediction = model.predict(np.array([5.1, 3.5, 1.4]))

给我:

(3,)
ValueError: Error when checking input: expected dense_13_input to have shape (3,) but got array with shape (1,)

我该如何解决这个问题?谢谢

标签: pythonmachine-learningkerasneural-networkartificial-intelligence

解决方案


你必须添加批次维度 (n_batch, feat_dim) ==> (1,3)

inp = Input((3))
x = Dense(10)(inp)
model = Model(inp, x)
X = np.array([5.1, 3.5, 1.4])

model.predict(X[None,:])

推荐阅读