【发布时间】:2021-12-18 23:41:08
【问题描述】:
在这里发帖是我最后的手段,因为我在网上找不到类似的东西。我训练了一个模型来将嵌入分类(一个简单的三层密集神经网络)。
现在我想使用经过训练的模型进行实时预测,但我发现如果我将整个 test 数据帧输入到模型中,就会得到元素号 @987654321 的预测@,并将其与我通过仅将 test 数据框的元素编号 i 输入到模型中得到的预测进行比较,我得到不同的结果。如果我解释得不够好,这是代码:
i = 522
y_pred = model.predict(X_test)
y_pred_2 = model.predict(X_test.iloc[[i]])
print (f'{np.argmax(y_pred[i])} {np.argmax(y_pred_2)}')
output: 8 5
就好像我的模型在一次运行中处理整个测试集与一次处理一行时的行为不同。我正在使用 pandas 作为输入数据。
编辑:更多信息,y_pred 和 y_pred_2 的输出形状分别为 (603, 10) 和 (1, 10),其中 10 是我拥有的类数。
两个预测的一些示例值,带有任意i:
y_pred[i]: array([1.3353945e-02, 2.8374636e-09, 1.4435661e-08, 3.4135045e-18,
7.7986561e-02, 3.7737598e-03, 2.0284578e-10, 2.7154891e-03,
9.0203673e-01, 1.3346069e-04], dtype=float32)
y_pred_2 = array([[1.1702824e-16, 1.6781385e-37, 2.5281618e-33, 0.0000000e+00,
2.3075200e-09, 1.0000000e+00, 9.9125501e-35, 6.2606384e-22,
5.8689110e-14, 2.3486194e-24]], dtype=float32)
【问题讨论】:
-
检查
y_pred和y_pred_2的形状 -
@PaulH 它们分别是 (603, 10) 和 (1,10),因为第一个包含对测试集中所有 603 行的预测
标签: python pandas machine-learning keras neural-network