diff --git a/prepare_train_model.py b/prepare_train_model.py index bf8667a..ab2c1e6 100644 --- a/prepare_train_model.py +++ b/prepare_train_model.py @@ -117,7 +117,6 @@ plt.xlabel('epoch') plt.legend(['loss'], loc='upper left') plt.show() - # Our losses are dropped significantly and the accuracy is almost 100%. For # testing our model, we randomly picked images over the internet and put it on # the test folder with a different class to test