UCDSC/split.py at main · Arnavadi19/UCDSC · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
splits_2020 = {

    'bloodmnist': [
        {'known': [6, 1, 3, 7], 'unknown': [0, 2, 4, 5]},
        {'known': [1, 6, 7, 2], 'unknown': [4, 3, 0, 5]},
        {'known': [2, 3, 6, 1], 'unknown': [0, 7, 4, 5]},
        {'known': [7, 1, 2, 3], 'unknown': [5, 4, 6, 0]},
        {'known': [0, 7, 4, 6], 'unknown': [1, 3, 5, 2]}
    ],
    'dermamnist': [
        {'known': [0, 2, 5, 6], 'unknown': [1, 3, 4]},
        {'known': [1, 5, 2, 4], 'unknown': [3, 6, 0]},
        {'known': [5, 1, 6, 4], 'unknown': [0, 3, 2]},
        {'known': [0, 3, 5, 6], 'unknown': [2, 4, 1]}
    ],
    'tissuemnist': [
        {'known': [0, 6, 3, 7], 'unknown': [1, 2, 4, 5]},
        {'known': [0, 6, 7, 4], 'unknown': [1, 3, 2, 5]},
        {'known': [4, 3, 0, 7], 'unknown': [6, 1, 2, 5]},
        {'known': [1, 3, 0, 6], 'unknown': [2, 4, 7, 5]},
        {'known': [6, 7, 3, 4], 'unknown': [0, 1, 2, 5]}
    ],
    'octmnist': [
        {'known': [3, 2], 'unknown': [0, 1]},
        {'known': [3, 0], 'unknown': [1, 2]},
        {'known': [3, 1], 'unknown': [0, 2]}
    ],
    'asc': [
        {'known': [0, 1, 2], 'unknown': [3, 4, 5]},
        {'known': [0, 3, 4], 'unknown': [1, 2, 5]},
        {'known': [1, 2, 5], 'unknown': [0, 3, 4]},
        {'known': [0, 2, 4], 'unknown': [1, 3, 5]}
    ]
}

"""Skin Conditions: Class Index → Condition Name
0 → Acne
1 → Carcinoma (Skin Cancer)
2 → Eczema
3 → Keratosis
4 → Milia
5 → Rosacea
"""