コード例 #1
0
ファイル: catalog.py プロジェクト: gcervantes8/OpenNMT-tf
 def __init__(self):
     super().__init__(
         inputter=inputters.MixedInputter(
             [
                 inputters.WordEmbedder(embedding_size=100),
                 inputters.CharConvEmbedder(
                     embedding_size=30,
                     num_outputs=30,
                     kernel_size=3,
                     stride=1,
                     dropout=0.5,
                 ),
             ],
             dropout=0.5,
         ),
         encoder=encoders.RNNEncoder(
             num_layers=1,
             num_units=400,
             bidirectional=True,
             dropout=0.5,
             residual_connections=False,
             cell_class=tf.keras.layers.LSTMCell,
         ),
         crf_decoding=True,
     )
コード例 #2
0
 def __init__(self):
     # pylint: disable=bad-continuation
     super(LstmCnnCrfTagger, self).__init__(
         inputter=inputters.MixedInputter([
             inputters.WordEmbedder(embedding_size=100),
             inputters.CharConvEmbedder(embedding_size=30,
                                        num_outputs=30,
                                        kernel_size=3,
                                        stride=1,
                                        dropout=0.5)
         ],
                                          dropout=0.5),
         encoder=encoders.RNNEncoder(num_layers=1,
                                     num_units=400,
                                     bidirectional=True,
                                     dropout=0.5,
                                     residual_connections=False,
                                     cell_class=tf.keras.layers.LSTMCell),
         crf_decoding=True)