@@ -27,41 +27,41 @@ def split_databunch(response, src):
27
27
28
28
if response_split ['validation' ]['method' ] == 'subsets' :
29
29
args = {
30
- 'train_size' : response_split ['validation' ]['subsets ' ]['train_size' ],
31
- 'valid_size' : response_split ['validation' ]['subsets ' ]['valid_size' ],
32
- 'seed' : response_split ['validation' ]['subsets ' ]['seed' ]
30
+ 'train_size' : response_split ['validation' ]['by_subsets ' ]['train_size' ],
31
+ 'valid_size' : response_split ['validation' ]['by_subsets ' ]['valid_size' ],
32
+ 'seed' : response_split ['validation' ]['by_subsets ' ]['seed' ]
33
33
}
34
34
35
35
if response_split ['validation' ]['method' ] == 'by_files' : # TODO: test it out
36
- args = {'valid_name' : response_split ['validation' ]['files ' ]['valid_names' ]}
36
+ args = {'valid_name' : response_split ['validation' ]['by_files ' ]['valid_names' ]}
37
37
38
38
if response_split ['validation' ]['method' ] == 'by_fname_file' :
39
39
args = {
40
- 'fname' : response_split ['validation' ]['fname_files ' ]['fname' ],
41
- 'path' : response_split ['validation' ]['fname_files ' ]['path' ]
40
+ 'fname' : response_split ['validation' ]['by_fname_files ' ]['fname' ],
41
+ 'path' : response_split ['validation' ]['by_fname_files ' ]['path' ]
42
42
}
43
43
44
44
if response_split ['validation' ]['method' ] == 'by_folder' :
45
45
args = {
46
- 'train' : response_split ['validation' ]['folder ' ]['train' ],
47
- 'valid' : response_split ['validation' ]['folder ' ]['valid' ]
46
+ 'train' : response_split ['validation' ]['by_folder ' ]['train' ],
47
+ 'valid' : response_split ['validation' ]['by_folder ' ]['valid' ]
48
48
}
49
49
# For tabular, same csv; for vision, csv with labels
50
50
if response_split ['validation' ]['method' ] == 'by_idx' :
51
51
df = pd .open_csv (response_split ['validation' ]['csv_name' ])
52
- valid_idx = range (len (df ) - response_split ['validation' ]['idx ' ]['valid_idx' ], len (df ))
52
+ valid_idx = range (len (df ) - response_split ['validation' ]['by_idx ' ]['valid_idx' ], len (df ))
53
53
args = {'valid_idx' : valid_idx }
54
54
55
55
if response_split ['validation' ]['method' ] == 'by_idxs' :
56
56
args = {
57
- 'train_idx' : response_split ['validation' ]['idxs ' ]['train_idx' ],
58
- 'valid_idx' : response_split ['validation' ]['idxs ' ]['valid_idx' ]
57
+ 'train_idx' : response_split ['validation' ]['by_idxs ' ]['train_idx' ],
58
+ 'valid_idx' : response_split ['validation' ]['by_idxs ' ]['valid_idx' ]
59
59
}
60
60
61
61
if response_split ['validation' ]['method' ] == 'by_list' :
62
62
args = {
63
- 'train' : response_split ['validation' ]['list ' ]['train' ],
64
- 'valid' : response_split ['validation' ]['list ' ]['valid' ]
63
+ 'train' : response_split ['validation' ]['by_list ' ]['train' ],
64
+ 'valid' : response_split ['validation' ]['by_list ' ]['valid' ]
65
65
}
66
66
67
67
if response_split ['validation' ]['method' ] == 'by_valid_func' :
0 commit comments