added .skip()

src/main.py

@@ -41,11 +41,13 @@ def main():
 
     # Defining dataset size as a function of the number of knots
     ninputs = len(knots) * len_db
-    dataset = dataset.take(int(ninputs)) # <-- takes after shuffle
+
+    datasets = dataset.skip(1) # <-- skips potential header 
+    datasets = datasets.take(int(ninputs)) # <-- takes after shuffle
 
     # Splitting in train test and validation according to chosen sizes
     train_dataset, test_dataset, val_dataset = split_train_test_validation(
-        dataset, int(ninputs * (0.9)), int(ninputs * (0.075)), int(ninputs * (0.025))
+        datasets, int(ninputs * (0.9)), int(ninputs * (0.075)), int(ninputs * (0.025))
     )
 
     # Starting training or testing according to user preferences