Ejemplo n.º 1
0
    def test_registry_has_builtin_seq2seq_encoders(self):

        assert Seq2SeqEncoder.by_name("gru")._module_class.__name__ == "GRU"
        assert Seq2SeqEncoder.by_name("lstm")._module_class.__name__ == "LSTM"
        assert Seq2SeqEncoder.by_name("rnn")._module_class.__name__ == "RNN"
Ejemplo n.º 2
0
 def test_registry_has_builtin_seq2seq_encoders(self):
     # pylint: disable=protected-access
     assert Seq2SeqEncoder.by_name('gru')._module_class.__name__ == 'GRU'
     assert Seq2SeqEncoder.by_name('lstm')._module_class.__name__ == 'LSTM'
     assert Seq2SeqEncoder.by_name('rnn')._module_class.__name__ == 'RNN'
Ejemplo n.º 3
0
 def test_registry_has_builtin_seq2seq_encoders(self):
     # pylint: disable=protected-access
     assert Seq2SeqEncoder.by_name('gru')._module_class.__name__ == 'GRU'
     assert Seq2SeqEncoder.by_name('lstm')._module_class.__name__ == 'LSTM'
     assert Seq2SeqEncoder.by_name('rnn')._module_class.__name__ == 'RNN'