should be starting from X, and Y X_{}_trainval, X_{}_test, Y_{}_trainval, Y_{}_test
(*args,**kwargs)
| 69 | |
| 70 | |
| 71 | def dataloader_graph(*args,**kwargs): |
| 72 | """ |
| 73 | should be starting from X, and Y |
| 74 | X_{}_trainval, X_{}_test, Y_{}_trainval, Y_{}_test |
| 75 | """ |
| 76 | temp_loader = {} |
| 77 | for name, input in kwargs.items(): |
| 78 | # First, format |
| 79 | temp_loader[name] = input |
| 80 | |
| 81 | # Second, combine trainval or test |
| 82 | temp_loader_trainval, temp_loader_test = [], [] |
| 83 | for key, val in temp_loader.items(): |
| 84 | load_name = key.split('_') |
| 85 | # train_val_dataset |
| 86 | if load_name[-1].endswith('trainval'): |
| 87 | temp_loader_trainval.append(val) |
| 88 | elif load_name[-1].endswith('test'): |
| 89 | temp_loader_test.append(val) |
| 90 | |
| 91 | |
| 92 | return temp_loader_trainval,temp_loader_test |
| 93 | |
| 94 | def k_fold_trainer(dataset,model,args): |
| 95 |