Last active
July 4, 2022 20:26
-
-
Save angeligareta/e3332c7a955dba8eaca71bf388d028c2 to your computer and use it in GitHub Desktop.
Method to split a tensorflow dataset (tf.data.Dataset) into train, validation and test splits
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| def get_dataset_partitions_tf(ds, ds_size, train_split=0.8, val_split=0.1, test_split=0.1, shuffle=True, shuffle_size=10000): | |
| assert (train_split + test_split + val_split) == 1 | |
| if shuffle: | |
| # Specify seed to always have the same split distribution between runs | |
| ds = ds.shuffle(shuffle_size, seed=12) | |
| train_size = int(train_split * ds_size) | |
| val_size = int(val_split * ds_size) | |
| train_ds = ds.take(train_size) | |
| val_ds = ds.skip(train_size).take(val_size) | |
| test_ds = ds.skip(train_size).skip(val_size) | |
| return train_ds, val_ds, test_ds |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
hello guys, am currently working on my tensorflow model to fit into a CNN model but then the problem am experiencing is that my kernel is not allow me to visualize my dataset. numpys are working prety good but when I run imshow my kernel says it's dead and will restart again,this problem is happening again and again even after restarting the kernel and running all the cells. Kindly, help me guys i'll appreciate so much when i get someone help solve this problem