MCPcopy Create free account
hub / github.com/diffusion-classifier/diffusion-classifier / get_target_dataset

Function get_target_dataset

diffusion/datasets.py:85–188  ·  view source on GitHub ↗

Get the torchvision dataset that we want to use. If the dataset doesn't have a class_to_idx attribute, we add it. Also add a file-to-class map for evaluation

(name: str, train=False, transform=None, target_transform=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 4

mainFunction · 0.90
mainFunction · 0.90
mainFunction · 0.90

Calls 7

get_test_datasetMethod · 0.95
get_classes_templatesFunction · 0.90
get_classnamesFunction · 0.90
ObjectNetBaseClass · 0.90
ImageNetV2Class · 0.85
ImageNetAClass · 0.85
MNISTClass · 0.85

Tested by

no test coverage detected