fastNLP.models.snli

class fastNLP.models.snli.ESIM(embed, hidden_size=None, num_labels=3, dropout_rate=0.3, dropout_embed=0.1)[源代码]

基类 fastNLP.models.BaseModel

别名 fastNLP.models.ESIM fastNLP.models.snli.ESIM

ESIM model的一个PyTorch实现 论文参见： https://arxiv.org/pdf/1609.06038.pdf
__init__(embed, hidden_size=None, num_labels=3, dropout_rate=0.3, dropout_embed=0.1)[源代码]
参数:
  • embed -- 初始化的Embedding
  • hidden_size (int) -- 隐藏层大小，默认值为Embedding的维度
  • num_labels (int) -- 目标标签种类数量，默认值为3
  • dropout_rate (float) -- dropout的比率，默认值为0.3
  • dropout_embed (float) -- 对Embedding的dropout比率，默认值为0.1
forward(words1, words2, seq_len1, seq_len2, target=None)[源代码]
参数:
  • words1 -- [batch, seq_len]
  • words2 -- [batch, seq_len]
  • seq_len1 -- [batch]
  • seq_len2 -- [batch]
  • target --
返回: