diff --git a/nli_rnn.py b/nli_rnn.py index 39e47e4..11d76ce 100755 --- a/nli_rnn.py +++ b/nli_rnn.py @@ -243,7 +243,7 @@ def predict(self, seq): vocab = [T, F] embedding = np.array([randvec(10) for _ in vocab]) - mod = ClassifierRNN(vocab=vocab, embedding=embedding, maxiter=50000) + mod = ClassifierRNN(vocab=vocab, embedding=embedding, maxiter=1000) mod.fit(copy.copy(train)) for x, y in train: