def test_mnist_fellowship():
    # no need to download the dataset again for this test (if it already exists)
    input_folder = os.path.join(dir_data, 'Data')
    MnistFellowship(path=input_folder, merge=False, download=False, train=True)
    check_task_sequences_files(scenario="mnist_fellowship",
                               folder=dir_data,
                               n_tasks=3,
                               dataset="mnist_fellowship",
                               train=True)
Example #2
0
def test_permutations_train(dataset, n_tasks):
    # no need to download the dataset again for this test (if it already exists)
    input_folder = os.path.join(dir_data, 'Data')
    Permutations(path=input_folder,
                 dataset=dataset,
                 tasks_number=n_tasks,
                 download=False,
                 train=True)
    check_task_sequences_files(scenario="Rotations",
                               folder=dir_data,
                               n_tasks=n_tasks,
                               dataset=dataset,
                               train=True)
def test_disjoint_vanilla_test(dataset, n_tasks):
    # no need to download the dataset again for this test (if it already exists)
    input_folder = os.path.join(dir_data, 'Data')
    Disjoint(path=input_folder,
             dataset=dataset,
             tasks_number=n_tasks,
             download=False,
             train=False)
    check_task_sequences_files(scenario="Disjoint",
                               folder=dir_data,
                               n_tasks=n_tasks,
                               dataset=dataset,
                               train=False)