Generate a dataset from the TFRecords files

2021-06-01 23:06:25 +02:00
parent 220c0482f1
commit d34e291085
2 changed files with 24 additions and 4 deletions
--- a/src/constants.py
+++ b/src/constants.py
@@ -1,3 +1,5 @@
 BASES = "ACGT"
 TRAIN_DATASET = "data/train_data.tfrecords"
 TEST_DATASET = "data/test_data.tfrecords"
+EPOCHS = 1000
+BATCH_SIZE = 256