Beispiel #1
0
 def test_registry_has_builtin_seq2vec_encoders(self):
     assert Seq2VecEncoder.by_name(u'cnn').__name__ == u'CnnEncoder'
     # pylint: disable=protected-access
     assert Seq2VecEncoder.by_name(u'gru')._module_class.__name__ == u'GRU'
     assert Seq2VecEncoder.by_name(
         u'lstm')._module_class.__name__ == u'LSTM'
     assert Seq2VecEncoder.by_name(u'rnn')._module_class.__name__ == u'RNN'
Beispiel #2
0
    def test_registry_has_builtin_seq2vec_encoders(self):
        assert Seq2VecEncoder.by_name("cnn").__name__ == "CnnEncoder"

        assert Seq2VecEncoder.by_name("gru")._module_class.__name__ == "GRU"
        assert Seq2VecEncoder.by_name("lstm")._module_class.__name__ == "LSTM"
        assert Seq2VecEncoder.by_name("rnn")._module_class.__name__ == "RNN"
 def test_registry_has_builtin_seq2vec_encoders(self):
     assert Seq2VecEncoder.by_name('cnn').__name__ == 'CnnEncoder'
     # pylint: disable=protected-access
     assert Seq2VecEncoder.by_name('gru')._module_class.__name__ == 'GRU'
     assert Seq2VecEncoder.by_name('lstm')._module_class.__name__ == 'LSTM'
     assert Seq2VecEncoder.by_name('rnn')._module_class.__name__ == 'RNN'