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 --
- 返回
-
training
: bool¶
-