From 24b45b45c6bbbb42443d4eb91ec39062a22039d0 Mon Sep 17 00:00:00 2001
From: Matthew Honnibal <honnibal+gh@gmail.com>
Date: Sat, 12 Aug 2017 17:15:28 -0500
Subject: [PATCH] Add test for beam update
---
spacy/tests/parser/test_neural_parser.py | 13 +++++++++++++
1 file changed, 13 insertions(+)
diff --git a/spacy/tests/parser/test_neural_parser.py b/spacy/tests/parser/test_neural_parser.py
index 42b55745f..30a6367c8 100644
--- a/spacy/tests/parser/test_neural_parser.py
+++ b/spacy/tests/parser/test_neural_parser.py
@@ -78,3 +78,16 @@ def test_predict_doc_beam(parser, tok2vec, model, doc):
parser(doc, beam_width=32, beam_density=0.001)
for word in doc:
print(word.text, word.head, word.dep_)
+
+
+def test_update_doc_beam(parser, tok2vec, model, doc, gold):
+ parser.model = model
+ tokvecs, bp_tokvecs = tok2vec.begin_update([doc])
+ d_tokvecs = parser.update_beam(([doc], tokvecs), [gold])
+ assert d_tokvecs[0].shape == tokvecs[0].shape
+ def optimize(weights, gradient, key=None):
+ weights -= 0.001 * gradient
+ bp_tokvecs(d_tokvecs, sgd=optimize)
+ assert d_tokvecs[0].sum() == 0.
+
+