Skip to content

Commit 311cd49

Browse files
committed
add docstring of Seq2SeqAutoEncoderUseWordHash
1 parent 435c9a6 commit 311cd49

1 file changed

Lines changed: 14 additions & 0 deletions

File tree

seq2vec/seq2seq_auto_encoder.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,20 @@ def _one_hot_encode_seq(seq, max_index):
3333

3434

3535
class Seq2SeqAutoEncoderUseWordHash(TrainableInterfaceMixin, BaseSeq2Vec):
36+
"""Hash words and feed to seq2seq auto-encoder.
37+
38+
Attributes
39+
----------
40+
max_index: int
41+
The length of input vector.
42+
43+
max_length: int
44+
The length of longest sequence.
45+
46+
latent_size: int
47+
The returned latent vector size after encoding.
48+
49+
"""
3650

3751
def __init__(self, max_index, max_length, latent_size=20):
3852
self.max_index = max_index

0 commit comments

Comments
 (0)