Skip to content

Two different types of neg_sampling for DIEN model. #1224

Answered by Ethan-TZ
Jeriousman asked this question in Q&A
Discussion options

You must be logged in to vote

@Jeriousman Hello, thanks for your attention to RecBole!
You don't need to set correct feature that will be entailed with '_list' to define which feature is sequence type.

  1. For DIEN, we implement speical interface to apply sequential data augmentation for both postive and negative sequential data. i.e., you just need to prepare the .inter in the form of <user,item,timestamp>,and we will generate the item_id_list and neg_item_id_list.
  2. All the datapoints are considered to be positive examples. Therefore, you only need to keep the positive example in the dataset and filter out negative examples. E.g., for ml-1m, just set:
    val_interval:
      rating: "[3,inf)"
    

Replies: 1 comment 5 replies

Comment options

You must be logged in to vote
5 replies
@Jeriousman
Comment options

@Ethan-TZ
Comment options

@Jeriousman
Comment options

@Ethan-TZ
Comment options

@Jeriousman
Comment options

Answer selected by Sherry-XLL
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
question Further information is requested
2 participants