python算法测试结果自动保存到excel表格的实现步骤

 更新时间:2023年12月20日 10:37:13   作者:Coding的叶子  
我们在进行算法评估是通常会针对每个样本的算法处理结果进行统计,例如每个样本正确预测数量、漏检数量和误检数量、精度等,本文小编将给大家介绍python算法测试结果自动保存到excel表格的实现步骤,感兴趣的朋友可以参考下

前言

我们在进行算法评估是通常会针对每个样本的算法处理结果进行统计,例如每个样本正确预测数量、漏检数量和误检数量、精度等。特别是,算法验证是比较关注错误识别的样本,把每个样本的预测结果写入excel表格,可以进行快速筛查和定位,并且结果更加直观。

1 python excel操作

这里会将算法结果保存成.xls格式的excel文件,所需的python库为xlwt。如果没有xlwt库,请使用pip进行安装。

pip install xlwt

1.1 excel文件创建

excel文件创建分为两步,第一步是定义表格的编码方式,第二步是为表格创建表单。

myWorkbook = xlwt.Workbook(encoding = 'utf-8')
mySheet = myWorkbook.add_sheet('Sheet1')

1.2 单元格赋值

单元格赋值的方式是采用write函数,第一个参数表示第row+1行,row=0表示第一行。第二个参数表示第col+1列,col=0表示第1列。参数label是要写入单元格的内容。

mySheet.write(row, col, label='测试')

1.3 表格保存

表格保存的文件后缀为.xls。

savepath = 'test.xls'
myWorkbook.save(savepath)

2 定义算法结果保存函数

函数名称为write_result_to_excel,定义如下。

def write_result_to_excel(files, fps, fns, prs, lbs, savepath='res.xls')
  • files:list,每个元素为样本名称字符串。
  • fps:list,每个元素为样本中误检目标数量。
  • fns:list,每个元素为样本中漏检目标数量。
  • prs:list,每个元素为样本中算法预测目标数量。
  • lbs:list,每个元素为样本中真实标签目标数量。
  • savepath:excel表格保存路径。

3 函数完整定义

可以根据自身需要对函数进行更改。

def write_result_to_excel(files, fps, fns, prs, lbs, savepath='res.xls'):
    myWorkbook = xlwt.Workbook(encoding = 'utf-8')
    mySheet = myWorkbook.add_sheet('Sheet1')
    mySheet.write(0, 0, label='样本名称')
    mySheet.write(0, 1, label='目标标签个数')
    mySheet.write(0, 2, label='预测标签个数')
    mySheet.write(0, 3, label='正确预测个数')
    mySheet.write(0, 4, label='错误预测个数')
    mySheet.write(0, 5, label='漏检目标个数')
    for i in range(len(files)):
        mySheet.write(i+1, 0, label=files[i])
        mySheet.write(i+1, 1, label=lbs[i])
        mySheet.write(i+1, 2, label=prs[i])
        mySheet.write(i+1, 3, label=lbs[i]-fns[i])
        mySheet.write(i+1, 4, label=fps[i])
        mySheet.write(i+1, 5, label=fns[i]) 
   
    fp = sum(fps)
    fn = sum(fns)
    pr = sum(prs)
    lb = sum(lbs)
 
    residual  = fn / lb * 100
    recall    = 100 - residual
    error     = fp / pr * 100
    precision = 100 - error
    accuracy  = recall * precision / 100
    mySheet.write(i+2, 0, label='总计')
    mySheet.write(i+2, 1, label=lb)
    mySheet.write(i+2, 2, label=pr)
    mySheet.write(i+2, 3, label=lb-fn)
    mySheet.write(i+2, 4, label=fp)
    mySheet.write(i+2, 5, label=fn)
 
    mySheet.write(i+3, 1, label='检出率(%)')
    mySheet.write(i+3, 2, label='漏检率(%)')
    mySheet.write(i+3, 3, label='准确率(%)')
    mySheet.write(i+3, 4, label='误检率(%)')
    mySheet.write(i+3, 5, label='精度(%)')
 
    mySheet.write(i+4, 1, label=recall)
    mySheet.write(i+4, 2, label=residual)
    mySheet.write(i+4, 3, label=precision)
    mySheet.write(i+4, 4, label=error)
    mySheet.write(i+4, 5, label=accuracy)
    myWorkbook.save(savepath)
    s = '目标标签个数:{0}\n预测标签个数:{1}\n正确预测个数:{2}\n错误预测个数:{3}\n漏检目标个数:{4}\n'.format(lb, pr, lb-fn, fp, fn)
    s += '检出率recall:{0}\n漏检率residual:{1}\n准确率precision:{2}\n误检率error:{3}\n精度accuracy:{4}'.format(recall, residual, precision, error, accuracy)
    print(s)

以上就是python算法测试结果自动保存到excel表格的实现步骤的详细内容,更多关于python算法测试结果保存excel的资料请关注脚本之家其它相关文章!

相关文章

  • Pytorch中torch.cat()函数举例解析

    Pytorch中torch.cat()函数举例解析

    一般torch.cat()是为了把多个tensor进行拼接而存在的,下面这篇文章主要给大家介绍了关于Pytorch中torch.cat()函数举例解析的相关资料,文中通过示例代码介绍的非常详细,需要的朋友可以参考下
    2022-12-12
  • Keras函数式(functional)API的使用方式

    Keras函数式(functional)API的使用方式

    这篇文章主要介绍了Keras函数式(functional)API的使用方式,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
    2023-02-02
  • django里面的forms模块详解

    django里面的forms模块详解

    这篇文章主要介绍了django里面的forms模块详解的相关资料,需要的朋友可以参考下
    2023-11-11
  • Python实现快速将pdf文件剪切成多个图片

    Python实现快速将pdf文件剪切成多个图片

    这篇文章主要为大家详细介绍了如何使用Python实现快速将pdf文件剪切成多个图片,文中的示例代码讲解详细,有需要的小伙伴可以跟随小编一起学习一下
    2024-01-01
  • django echarts饼图数据动态加载的实例

    django echarts饼图数据动态加载的实例

    今天小编就为大家分享一篇django echarts饼图数据动态加载的实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-08-08
  • Python字典生成式、集合生成式、生成器用法实例分析

    Python字典生成式、集合生成式、生成器用法实例分析

    这篇文章主要介绍了Python字典生成式、集合生成式、生成器用法,结合实例形式分析了Python字典生成式、集合生成式、生成器相关原理、使用技巧与操作注意事项,需要的朋友可以参考下
    2020-01-01
  • python进阶_浅谈面向对象进阶

    python进阶_浅谈面向对象进阶

    下面小编就为大家带来一篇python进阶_浅谈面向对象进阶。小编觉得挺不错的,现在就分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2017-08-08
  • 使用Python提取文本中含有特定字符串的方法示例

    使用Python提取文本中含有特定字符串的方法示例

    这篇文章主要给大家介绍了关于如何使用Python提取文本中含有特定字符串的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2020-12-12
  • 简单示例入门了解WxPython教程

    简单示例入门了解WxPython教程

    这篇文章主要通过演示简单示例带大家入门了解WxPython,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-11-11
  • 一篇文章带你了解python标准库--time模块

    一篇文章带你了解python标准库--time模块

    下面小编就为大家带来一篇python模块之time模块。小编觉得挺不错的,现在就分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2021-08-08

最新评论