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]