Skip to content
Snippets Groups Projects
Commit 6ad28046 authored by Djordje Mihajlovic's avatar Djordje Mihajlovic
Browse files

added .skip()

parent 1e4d7c61
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment