Skip to content

关于DICE的实现 #3

@zhw741852

Description

@zhw741852

DICE的实现中为什么interest建模是在O1(i的流行度大于j的流行度)上进行的,原论文在O2上进行(i的流行度小于j的流行度)

user_embedding = self.embed_user[user_O1, 0:DICE_size]
item_i_embedding = self.embed_item[item_i_O1, 0:DICE_size]
item_j_embedding = self.embed_item[item_j_O1, 0:DICE_size]
prediction_i = (user_embedding * item_i_embedding).sum(
    dim=-1)
prediction_j = (user_embedding * item_j_embedding).sum(
    dim=-1)
loss_interest = - \
    (prediction_i - prediction_j).sigmoid().log().sum()

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions