fixed test set selection

Novartis · Jul 11, 2023 · 1b99b67 · 1b99b67
1 parent 06934f5
commit 1b99b67
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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])