From 1b99b6760de3ecee7860ee6587e50248d85ade0e Mon Sep 17 00:00:00 2001 From: Michal Pikusa Date: Tue, 11 Jul 2023 20:00:43 +0200 Subject: [PATCH] fixed test set selection --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 178ed0e..2ffe51f 100644 --- a/train.py +++ b/train.py @@ -135,7 +135,7 @@ def main( dataset = create_datasets(compounds, profiles) train_len = int(len(dataset)*0.9) train_set = dataset.take(train_len) - valid_set = dataset.take(train_len) + valid_set = dataset.skip(train_len) train(train_set, valid_set, epochs, compounds.shape[1], profiles.shape[1])