2.6 Implementation



We are using a dataset of Cats and Dogs. You can use the method I used, or just skip some of the code by extracting the files.


The implementation is above. The model has an accuracy of about 87%, not bad for 10 epochs. It might take a while to train because it has to train on images.


The loss and accuracy, in that order, of my model are shown. Your model should look roughly the same.
notion image
notion image

Previous Section

2.5 Reducing Overfitting
Copyright © 2021 Code 4 Tomorrow. All rights reserved. The code in this course is licensed under the MIT License. If you would like to use content from any of our courses, you must obtain our explicit written permission and provide credit. Please contact classes@code4tomorrow.org for inquiries.