我已经构建了一个Keras模型,用于从原始文本输入中提取信息。我的准确率达到了0.9869。我如何知道是哪些训练数据导致了准确率的下降?我在下面粘贴了我使用的代码。
回答:
我认为最简单的方法如下:用训练数据训练模型,然后在训练数据上进行预测,并查看预测错误的训练样本。
代码示例:
model.fit(x_nyha, y_nyha, batch_size=batch_size, epochs=num_epochs, verbose=1)prediction = np.round(model.predict(x_nyha))wrong_predictions = x_nyha[prediction != y_nyha]
这样,wrong_predictions
中就包含了预测错误的行数据。