コード例 #1
0
ファイル: main.py プロジェクト: henritoi/gan-ai-project
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
コード例 #2
0
ファイル: main.py プロジェクト: henritoi/gan-ai-project
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