From 900c78b11e5da28e4d3cf5745bd5d477aca90f63 Mon Sep 17 00:00:00 2001 From: Tobias Fischer Date: Thu, 6 Aug 2020 13:02:15 +1000 Subject: [PATCH] Update train_model.py --- rt_gene_model_training/pytorch/train_model.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/rt_gene_model_training/pytorch/train_model.py b/rt_gene_model_training/pytorch/train_model.py index 433bb50..9db6853 100644 --- a/rt_gene_model_training/pytorch/train_model.py +++ b/rt_gene_model_training/pytorch/train_model.py @@ -173,13 +173,13 @@ def test_dataloader(self): _train_subjects.append([1, 2, 8, 10, 5, 6, 11, 12, 13]) _train_subjects.append([3, 4, 7, 9, 5, 6, 11, 12, 13]) # validation set is always subjects 14, 15 and 16 - _valid_subjects.append([14, 15, 16]) - _valid_subjects.append([14, 15, 16]) - _valid_subjects.append([14, 15, 16]) + _valid_subjects.append([0, 14, 15, 16]) + _valid_subjects.append([0, 14, 15, 16]) + _valid_subjects.append([0, 14, 15, 16]) # test subjects - _test_subjects.append([0, 5, 6, 11, 12, 13]) - _test_subjects.append([0, 3, 4, 7, 9]) - _test_subjects.append([0, 1, 2, 8, 10]) + _test_subjects.append([5, 6, 11, 12, 13]) + _test_subjects.append([3, 4, 7, 9]) + _test_subjects.append([1, 2, 8, 10]) else: _train_subjects.append([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16]) _valid_subjects.append([0]) # Note that this is a hack and should not be used to get results for papers