在pytorch上绘制CIFAR10数据集类别的频率

如何解决在pytorch上绘制CIFAR10数据集类别的频率

我想在图表上可视化cifar10数据集的每个类的图像数据数量，我尝试在pytorch上实现此功能，但无法获得令人满意的结果。 谁能帮我在pytorch上绘制CIFAR10数据集类的频率。

这是我尝试过的一种实现方式，但是在将标签与类索引进行比较时，我总是遇到错误，并且plot函数似乎运行得很好

def classes_freq():
    for images,labels in trainloader:
          #print(labels.shape)
      
          images = images.to(device)
          labels = labels.to(device)
          count = [0]*10
          for i,e in enumerate(classes):
            if (labels.item() == i):
              count[i]+=1
    return count

classes_freq()

import matplotlib.pyplot as plt

def plot(y_units):

  x_units = ['plane','car','bird','cat','deer','dog','frog','horse','ship','truck']

  plt.bar(x_units,y_units,width=0.8,color=['red','blue'])

  plt.xlabel('x - axis')
  plt.ylabel('y - axis')
  plt.title('classes frequency')

  plt.show()

#plot([10,20,23,50,60,70,80,90,10,45])