1. 首页
  2. 编程语言
  3. Python
  4. pytorch 多分类问题,计算百分比操作

pytorch 多分类问题,计算百分比操作

上传者: 2021-05-23 19:59:19上传 PDF文件 102.79 KB 热度 10次

二分类或分类问题,网络输出为二维矩阵:批次x几分类,最大的为当前分类,标签为one-hot型的二维矩阵:批次x几分类计算百分比有numpy和pytorch两种实现方案实现,都是根据索引计算百分比,以下为具体二分类实现过程。最近帮一个人做了一个多分类画auc曲线的东西,不过最后那个人不要了,还被说了一顿,心里很是不爽,anyway,我写代码的还是要继续写代码的,所以我准备把我修改的代码分享开来,供大家研究学习。处理的数据大改是这种xlsx文件:导入基础的pandas和keras处理函数。计算每个类别的fpr和tpr计算macro auc画图展示上述的代码是在jupyter中运行的,所以是分开的

用户评论