Fix training when hist_size==0

2017-10-05 21:52:28 -05:00 · 2017-10-05 21:52:28 -05:00 · fc06b0a333
parent e25ffcb11f
commit fc06b0a333
1 changed files with 2 additions and 0 deletions
--- a/spacy/_ml.py
+++ b/spacy/_ml.py
@ -257,6 +257,8 @@ class Embed(Model):

 def HistoryFeatures(nr_class, hist_size=8, nr_dim=8):
    '''Wrap a model, adding features representing action history.'''
+    if hist_size == 0:
+        return layerize(noop())
    embed_tables = [Embed(nr_dim, nr_class, column=i, name='embed%d')
                    for i in range(hist_size)]
    embed = concatenate(*embed_tables)