Data Arrangement

Data format

data(input path)
|    config.json
|    iteration_structure.json
|
└───Model
│   │   model.py
│   │
│   └───Epoch_1
│       │   index.json
│       │   subject_model.pth
│   └───Epoch_2
|       |   ...
│   
└───Training_data
|   │   training_dataset_data.pth
|   │   training_dataset_label.pth
│   
└───Testing_data
│   │   testing_dataset_data.pth
│   │   testing_dataset_label.pth

model.py

Model wrapper must have those two functions.

net = model.resnet18()
net.feature_funtion()
net.prediction_function()