Skip to content

Use Criteo Dataset #1076

Answered by Sherry-XLL
DX12321 asked this question in Q&A
Discussion options

You must be logged in to vote

Criteo 数据集运行 Context 模型的示例配置文件如下:

# dataset config
field_separator: "\t"
seq_separator: " "
USER_ID_FIELD: ~
ITEM_ID_FIELD: ~
LABEL_FIELD: label

load_col: 
    inter: '*'

fill_nan: True
normalize_all: True


# training and evaluation
epochs: 500
train_batch_size: 4096
eval_args:
  mode: labeled
  group_by: ~
valid_metric: AUC
metrics: ['AUC', 'LogLoss']

# model
embedding_size: 10

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
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