How to separate graphs by classes in dataset 'BACE' #5463

jiaruHithub · 2022-09-17T10:16:53Z

jiaruHithub
Sep 17, 2022

How to separate graphs by classes in dataset 'BACE'?
Thanks!

Answered by Padarn

You mean to extract the list of batch indexes for each class? If so you could just do something simple like

from collections import defaultdict
classes = {}
for i, d in enumerate(ds):
    classes[d.y.item()].append(i)

Padarn · 2022-09-18T00:07:10Z

You mean to extract the list of batch indexes for each class? If so you could just do something simple like

from collections import defaultdict
classes = {}
for i, d in enumerate(ds):
    classes[d.y.item()].append(i)

0 replies