今天写代码的时候,要统计一下标签数据里出现的类别总数和要分类的分类数是不是一致的。我的做法是把tensor类型的数据转变成list,然后用Counter函数做统计。代码如下:
from collections import Counter
List_counter = Counter(List1)
#List1就是待统计的数据,是一维的列表。生成的List_counter是一个字典,键是数据,
#对应的值是数据出现的频率
在做这个统计的时候,突然发现,我的数据是float的类型,这是不应该出现的,因为标签数据在处理的时候都是整型数据。经过一番查找后,发现是tensor.numpy()返回值数据类型的原因。这个函数的返回值是float类型的