Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- dog_images = [image for image in os.listdir(DATA_DIR) if 'dog' in image]
- cat_images = [image for image in os.listdir(DATA_DIR) if 'cat' in image]
- dog_train = dog_images[:int(len(dog_images)*0.7)]
- dog_test = dog_images[int(len(dog_images)*0.7):]
- cat_train = cat_images[:int(len(cat_images)*0.7)]
- cat_test = cat_images[int(len(cat_images)*0.7):]
Add Comment
Please, Sign In to add comment