Skip to content

Commit 93dc5c4

Browse files
committed
Update docstring about cross validation fold usage
Signed-off-by: MaybeRichard <[email protected]>
1 parent e0bba33 commit 93dc5c4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

generation/maisi/maisi_train_controlnet_tutorial.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -166,7 +166,7 @@
166166
" \"image\": \"tr_image_001_emb.nii.gz\", # relative path to the image embedding file\n",
167167
" # relative path to the combined label (pseudo whole-body segmentation mask + ROI mask) file\n",
168168
" \"label\": \"tr_label_001.nii.gz\",\n",
169-
" \"fold\": 0, # fold index for cross validation, fold 0 is used for training\n",
169+
" \"fold\": 0, # fold index for cross validation. If the dataset item's fold value is equal to the fold value in config_maisi_controlnet_train.json, then it is used for validation. Otherwise, it is used for training. In the current parameters, fold 0 is for validation and fold 1 is for training.\n",
170170
" \"dim\": sim_dim, # the dimension of image\n",
171171
" \"spacing\": [1.5, 1.5, 1.5], # the spacing of image\n",
172172
" \"top_region_index\": [0, 1, 0, 0], # the top region index of the image\n",

0 commit comments

Comments
 (0)