📅  最后修改于: 2020-12-11 05:06:30             🧑  作者: Mango
预测看不见的数据中的数字非常容易。您只需将模型传递给包含未知数据点的向量来调用模型的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
现在,当您对模型进行了令人满意的培训后,我们将其保存以备将来使用。