def get_configurations()

in scripts/split_dataset.py [0:0]

• 32 lines of code
• 10 McCabe index (conditional complexity)

def get_configurations(neighborhoods):
    train_configurations = list()
    valid_configurations = list()
    test_configurations = list()

    for neighborhood in neighborhoods:
        cnt = 0
        for minimum_x in range(boundaries[neighborhood][0]+1):
            for minimum_y in range(boundaries[neighborhood][1]+1):
                min_x = minimum_x*2
                min_y = minimum_y*2

                boundary_config = list()
                for i in range(4):
                    for j in range(4):
                        x = min_x + i
                        y = min_y + j

                        config = {'neighborhood': neighborhood,
                                  'target_location': [x, y, 0],
                                  'boundaries': [min_x, min_y, min_x + 3, min_y + 3]}

                        boundary_config.append(config)

                        if not (minimum_y >= boundaries[neighborhood][1]-1 and minimum_x < 3):
                            key2set[(neighborhood, minimum_x, minimum_y)] = 'train'
                            train_configurations.append(config)
                            cnt += 1
                        elif minimum_y == boundaries[neighborhood][1] and minimum_x < 2:
                            key2set[(neighborhood, minimum_x, minimum_y)] = 'test'
                            test_configurations.append(config)
                            cnt += 1
                        else:
                            key2set[(neighborhood, minimum_x, minimum_y)] = 'valid'
                            valid_configurations.append(config)
                            cnt += 1
    return train_configurations, valid_configurations, test_configurations