2024-03-28 10:52:46 发布
网友
如果我有如下数据集:
image_datasets['train'] = datasets.ImageFolder(train_dir, transform=train_transforms)
如何通过编程确定数据集中类或唯一标签的数量
使用:
len(image_datasets['train'].classes)
.classes返回一个列表
.classes
如果您的数据类型是张量,则可以使用:
import torch n_classes = len(torch.unique(Your_Target_Vector))
使用:
.classes
返回一个列表如果您的数据类型是张量,则可以使用:
import torch n_classes = len(torch.unique(Your_Target_Vector))
相关问题 更多 >
编程相关推荐