如何在pytorch中对数据集进行排序

作者: 不浪漫罪名
发布时间: 2024-12-16 01:01:09 (1月前)
转自：

2 条回复

0#
回复此人
部落用户 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 没有通用的方法可以有效地执行此操作，因为数据集类只实现了一个 <code> __getitem__ </code> 和 <code> __len__ </code> 方法，并且不一定具有关于标签的任何“存储”信息。 </p> <P> 在的情况下 <a href="https://pytorch.org/docs/stable/_modules/torchvision/datasets/mnist.html#MNIST" rel="nofollow noreferrer"> MNIST数据集 </A> 但是，您可以从标签列表中对数据集进行排序。 </p> <P> 例如，当您要列出标签为5的索引时。 </p> <pre> <code> mnist = torchvision.datasets.mnist.MNIST("/") labels = mnist.train_labels fives = (labels == 5).nonzero() </code> </pre> </DIV>

编辑

登录后才能参与评论