NCE
该接口用于构建 NCE
类的一个可调用对象,具体用法参照 代码示例
。其中实现了 NCE
损失函数的功能,其默认使用均匀分布进行抽样,计算并返回噪音对比估计( noise-contrastive estimation training loss)。更多详情请参考:
参数:
num_total_classes (int) - 所有样本中的类别的总数。
dim (int) - 输入的维度(一般为词嵌入的维度)。
param_attr (ParamAttr, 可选) - 指定权重参数属性的对象。默认值为None,表示使用默认的权重参数属性。具体用法请参见 ParamAttr 。
num_neg_samples (int, 可选) - 负样本的数量。默认值:10。
custom_dist (float[], 可选) – float[] 类型的数据,并且它的长度为 。如果采样器类别为
custom_dist
,则使用此参数。custom_dist[i]是第i个类别被取样的概率。默认值:Noneseed (int, 可选) – 采样器使用的随机种子。默认值:0。
is_sparse (bool, 可选) – 指明是否使用稀疏更新,如果为True,
返回:无
代码示例
weight
本层的可学习参数,类型为 Parameter