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
