16）用微调的模型进行预测猫狗；keras；deeplizard

实训部分

代码

其实就一行

predictions = model.predict(x=test_batches, steps=len(test_batches), verbose=0)


cm = confusion_matrix(y_true=test_batches.classes, y_pred=np.argmax(predictions, axis=-1))
cm_plot_labels = ['cat','dog']
plot_confusion_matrix(cm=cm, classes=cm_plot_labels, title='Confusion Matrix')

结果

参考：

vid