diff --git a/README.md b/README.md index 330757a..c1930c1 100644 --- a/README.md +++ b/README.md @@ -16,13 +16,16 @@ The model is trained on the training set using the Adam optimizer and binary cro ## Requirements -> Python 3.6 or later +> Python 3.6 or later + > TensorFlow 2.0 or later + > Keras 2.0 or later ## Usage To train the model, run the the following Jupyter notbook: + `imdb.ipynb` ## Results