Adding bag_of_embeddings as an alternate name for the boe encoder (Is…

…sue #2268) (#2521)
allenai · Feb 18, 2019 · 2f2b481 · 2f2b481
1 parent 9f87aa5
commit 2f2b481
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/allennlp/modules/seq2vec_encoders/boe_encoder.py b/allennlp/modules/seq2vec_encoders/boe_encoder.py
@@ -6,6 +6,7 @@
 from allennlp.nn.util import get_lengths_from_binary_sequence_mask
 
 @Seq2VecEncoder.register("boe")
+@Seq2VecEncoder.register("bag_of_embeddings")
 class BagOfEmbeddingsEncoder(Seq2VecEncoder):
     """
     A ``BagOfEmbeddingsEncoder`` is a simple :class:`Seq2VecEncoder` which simply sums the embeddings