python 的topk算法实例

2023-07-27 18:00:14 272

#!conding:utf-8 defquick_index(array,start,end): left,right=start,end key=array[left] whileleftkey: right-=1 array[left]=array[right] whileleft

补充知识：pythonnumpy求top-kaccuracy指标

top-kacc表示在多分类情况下取最高的k类得分的label，与真实值匹配，只要有一个labelmatch，结果就是True。

如对于一个有5类的多分类任务

a_real=1
a_pred=[0.02,0.23,0.35,0.38,0.02]

#top-1
a_pred_label=3match=False
#top-3
a_pred_label_list=[1,2,3]match=True

对于top-1accuracy

sklearn.metrics提供accuracy的方法，能够直接计算得分，但是对于topk-acc就需要自己实现了：

#5类：0，1，2，3，4
importnumpyasnp
a_real=np.array([[1],[2],[1],[3]])
#用随机数代替分数
random_score=np.random.rand((4,5))
a_pred_score=random_score/random_score.sum(axis=1).reshape(random_score.shape[0],1)

k=3#top-3
#以下是计算方法
max_k_preds=a_pred_score.argsort(axis=1)[:,-k:][:,::-1]#得到top-klabel
match_array=np.logical_or.reduce(max_k_preds==a_real,axis=1)#得到匹配结果
topk_acc_score=match_array.sum()/match_array.shape[0]

以上这篇python的topk算法实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。

声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。

python 的topk算法实例

热门推荐

随机推荐