pytorch图像分类实战 之测试集评估模型指标

前言

第五次任务是使用测试集数据对模型进行评价。代码教程：https://github.com/TommyZihao/Train_Custom_Dataset 。

实验效果

混淆矩阵

PR曲线

ROC曲线

总结

本次实验主要是怎样使用指标对模型进行评估，最后展示出包括混淆矩阵、总体准确率、PR曲线、ROC曲线等指标进行模型评估，数据是使用测试集数据。通过本次学习，我复习回顾了模型评估指标相关的知识，并通过动手实验加深印象，本次学习令我受益匪浅。