Deep Learning With Keras 简明教程
Predicting on Test Data
预测未知数据中的数字很容易。您只需调用 model 的 predict_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
现在,您已经对模型进行了满意的训练,我们将保存它以备将来使用。