def __get_dataset_selection(): downloaded_datasets = __get_downloaded_datasets() input_helper = InputHelper(help='\nSelect training dataset to be used:', options=downloaded_datasets, default=0) dataset = input_helper.get_output() return dataset
def __get_command_selection(): input_helper = InputHelper(help='\nSelect command to be used:', options=['train', 'test', 'list'], default=0) command = input_helper.get_output() return command