Skip to content

Commit 877e7ac

Browse files
committed
Update docstring for AverageEmbeddingInputLayer
Remove irrelevant references, and clarify that it's not just used for FastText
1 parent d0b99e6 commit 877e7ac

1 file changed

Lines changed: 6 additions & 4 deletions

File tree

tensorlayer/layers.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -652,7 +652,10 @@ def __init__(
652652

653653

654654
class AverageEmbeddingInputLayer(Layer):
655-
"""The :class:`AverageEmbeddingInputlayer` class is for FastText Embedding for sentence classification, see `[1] <http://arxiv.org/abs/1607.01759>`_.
655+
""":class:`AverageEmbeddingInputlayer` averages over embeddings of inputs.
656+
657+
:class:`AverageEmbeddingInputlayer` can be used as the input layer
658+
for models like DAN[1] and FastText[2].
656659
657660
Parameters
658661
------------
@@ -665,9 +668,8 @@ class AverageEmbeddingInputLayer(Layer):
665668
666669
References
667670
------------
668-
- [1] Joulin, A., Grave, E., Bojanowski, P., & Mikolov, T. (2016). `Bag of Tricks for Efficient Text Classification. <http://arxiv.org/abs/1607.01759>`_
669-
- [2] Recht, B., Re, C., Wright, S., & Niu, F. (2011). `Hogwild: A Lock-Free Approach to Parallelizing Stochastic Gradient Descent. <https://arxiv.org/abs/1106.5730>`_ In NPIS 2011 (pp. 693–701).
670-
- [3] `TensorFlow Candidate Sampling <https://www.tensorflow.org/api_guides/python/nn#Candidate_Sampling>`_
671+
- [1] Iyyer, M., Manjunatha, V., Boyd-Graber, J., & Daum’e III, H. (2015). Deep Unordered Composition Rivals Syntactic Methods for Text Classification. In Association for Computational Linguistics.
672+
- [2] Joulin, A., Grave, E., Bojanowski, P., & Mikolov, T. (2016).`Bag of Tricks for Efficient Text Classification. <http://arxiv.org/abs/1607.01759>`_
671673
"""
672674
def __init__(
673675
self, inputs, vocabulary_size, embedding_size,

0 commit comments

Comments
 (0)