根据测试数据进行预测
预测未知数据中的数字非常简单。您只需通过将模型的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
现在,由于您已经令人满意地训练了模型,我们将保存它以备将来使用。