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个类别被取样的概率。默认值:None

    • seed (int, 可选) – 采样器使用的随机种子。默认值:0。

    • is_sparse (bool, 可选) – 指明是否使用稀疏更新,如果为True,

    返回:无

    代码示例

    weight

    本层的可学习参数,类型为 Parameter