Exemplo n.º 1
0
 def serve_interaction_graph(subgraph):
     MLPSoftmax(user=None,
                item=subgraph['seq_item_vec'],
                seq_len=subgraph['seq_len'],
                max_seq_len=max_seq_len,
                dims=[dim_item_embed, total_items],
                l2_reg=l2_reg_mlp,
                train=False,
                subgraph=subgraph,
                scope='MLPSoftmax')
Exemplo n.º 2
0
 def train_interaction_graph(subgraph):
     MLPSoftmax(user=None,
                item=subgraph['seq_item_vec'],
                seq_len=subgraph['seq_len'],
                max_seq_len=max_seq_len,
                dims=[dim_item_embed, total_items],
                l2_reg=l2_reg_mlp,
                labels=subgraph['label'],
                dropout=dropout,
                train=True,
                subgraph=subgraph,
                scope='MLPSoftmax')
Exemplo n.º 3
0
 def serve_interaction_graph(subgraph):
     MLPSoftmax(user=subgraph['user_vec'],
                item=subgraph['seq_item_vec'],
                seq_len=subgraph['seq_len'],
                max_seq_len=max_seq_len,
                dims=[
                    dim_user_embed['total'] + dim_item_embed['total'],
                    item_dict['id']
                ],
                l2_reg=l2_reg_mlp,
                train=False,
                subgraph=subgraph,
                scope='MLPSoftmax')
Exemplo n.º 4
0
 def train_interaction_graph(subgraph):
     MLPSoftmax(user=subgraph['user_vec'],
                item=subgraph['seq_item_vec'],
                seq_len=subgraph['seq_len'],
                max_seq_len=max_seq_len,
                dims=[
                    dim_user_embed['total'] + dim_item_embed['total'],
                    item_dict['id']
                ],
                l2_reg=l2_reg_mlp,
                labels=subgraph['label'],
                dropout=dropout,
                train=True,
                subgraph=subgraph,
                scope='MLPSoftmax')
Exemplo n.º 5
0
 def serve_interaction_graph(subgraph):
     MLPSoftmax(
         user=subgraph["user_vec"],
         item=subgraph["seq_item_vec"],
         seq_len=subgraph["seq_len"],
         max_seq_len=max_seq_len,
         dims=[
             dim_user_embed["total"] + dim_item_embed["total"],
             item_dict["id"]
         ],
         l2_reg=l2_reg_mlp,
         train=False,
         subgraph=subgraph,
         scope="MLPSoftmax",
     )
Exemplo n.º 6
0
 def train_interaction_graph(subgraph):
     MLPSoftmax(
         user=subgraph["user_vec"],
         item=subgraph["seq_item_vec"],
         seq_len=subgraph["seq_len"],
         max_seq_len=max_seq_len,
         dims=[
             dim_user_embed["total"] + dim_item_embed["total"],
             item_dict["id"]
         ],
         l2_reg=l2_reg_mlp,
         labels=subgraph["label"],
         dropout=dropout,
         train=True,
         subgraph=subgraph,
         scope="MLPSoftmax",
     )