Embedding嵌入使用

• Embedding定义
  • 常见定义
• Embedding处理
  • Embedding Engine
  • 注意
• 特征准入与特征过滤
  • 特征准入
  • 特征淘汰

简单示例

def get_emb_conf():
    emb_options = {}
    emb_options["item_id"] = EmbeddingOption(
       embedding_dim=8,
       shared_name="item_id",
       combiner="mean",
       device=torch.device("cuda"),
    )
    emb_options["user_id"] = EmbeddingOption(
       embedding_dim=8,
       shared_name="user_id",
       combiner="mean",
       device=torch.device("cuda"),
    )
    emb_options["rate"] = EmbeddingOption(
       embedding_dim=8,
       shared_name="item_id",
       combiner="mean",
       device=torch.device("cuda"),
    )
return emb_options