diff --git a/notebooks/bi_lstm_sentiment_classifier.ipynb b/notebooks/bi_lstm_sentiment_classifier.ipynb index 8db3493..5dd5c2a 100644 --- a/notebooks/bi_lstm_sentiment_classifier.ipynb +++ b/notebooks/bi_lstm_sentiment_classifier.ipynb @@ -72,12 +72,12 @@ "output_dir = 'model_output/biLSTM'\n", "\n", "# training:\n", - "epochs = 6\n", + "epochs = 656423\n", "batch_size = 128\n", "\n", "# vector-space embedding: \n", "n_dim = 64 \n", - "n_unique_words = 10000 \n", + "n_unique_words = 1000000\n", "max_review_length = 200 # doubled!\n", "pad_type = trunc_type = 'pre'\n", "drop_embed = 0.2 \n",