Deep Learning With Keras 简明教程

Predicting on Test Data

预测未知数据中的数字很容易。您只需调用 modelpredict_classes 方法,方法是将其传递给包含未知数据点的向量。

predictions = model.predict_classes(X_test)

该方法调用在一个向量中返回预测,这个向量可以针对实际值进行 0 和 1 的测试。这是通过以下两个语句完成的:

correct_predictions = np.nonzero(predictions == y_test)[0]
incorrect_predictions = np.nonzero(predictions != y_test)[0]

最后,我们将使用以下两个程序语句打印预测正确和不正确的计数:

print(len(correct_predictions)," classified correctly")
print(len(incorrect_predictions)," classified incorrectly")

运行代码时,您将获得以下输出:

9837 classified correctly
163 classified incorrectly

现在,您已经对模型进行了满意的训练,我们将保存它以备将来使用。