def get_create_inputs(dataset_name: str, is_train: bool, epochs: int):
options = {
'mnist': lambda: create_inputs_mnist(is_train),
'fashion_mnist': lambda: create_inputs_mnist(is_train),
'smallNORB': lambda: create_inputs_norb(is_train, epochs)
}
return options[dataset_name]
def get_create_inputs(dataset_name: str, is_train: bool, epochs: int):
options = {'mnist': lambda: create_inputs_mnist(is_train),
'asl': lambda: create_inputs_asl(is_train),
'italy': lambda: create_inputs_italy(is_train),
'fashion_mnist': lambda: create_inputs_mnist(is_train),
'smallNORB': lambda: create_inputs_norb(is_train, epochs),
'cifar10': lambda: create_inputs_cifar10(is_train)}
return options[dataset_name]
def get_create_inputs(dataset_name, is_train, epochs, chunk_id=0):
options = {
'mnist': lambda: create_inputs_mnist(is_train),
'landmark': lambda: create_inputs_landmark(is_train, chunk_id=chunk_id)
}
return options[dataset_name]
