Commit e02f8ba

author

Stefan Knegt

authored

Update README.md

1 parent e425f7a commit e02f8baCopy full SHA for e02f8ba

File tree

-1

lines changed

-1

lines changed

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -46,7 +46,7 @@ from load_LIDC_data import LIDC_IDRI`
`46`	`46`	`dataset = LIDC_IDRI(dataset_location = 'insert_path_here')`
`47`	`47`	`dataset_size = len(dataset)`
`48`	`48`	`indices = list(range(dataset_size))`
`49`		`-split = int(np.floor(test_split * dataset_size))`
	`49`	`+split = int(np.floor(0.1 * dataset_size))`
`50`	`50`	`np.random.shuffle(indices)`
`51`	`51`	`train_indices, test_indices = indices[split:], indices[:split]`
`52`	`52`	`train_sampler = SubsetRandomSampler(train_indices)`

Comments

(0)