根据测试数据进行预测

预测未知数据中的数字非常简单。您只需通过将模型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 correct")
print(len(incorrect_predictions)," classified invalidly")

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

9837 classified correct
163 classified invalidly

现在,由于您已经令人满意地训练了模型,我们将保存它以备将来使用。