def __init__()

in Seq2SeqNMT/seq2seq_nmt.py [0:0]

• 5 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self, input_size, hidden_size):
        super(EncoderRNN, self).__init__()
        self.hidden_size = hidden_size

        self.embedding = nn.Embedding(input_size, hidden_size)
        self.gru = nn.GRU(hidden_size, hidden_size)