pdn_on_taobao.config

# Note: this is just a demo using faked data.

train_input_path: "data/test/tb_data/taobao_pdn_fake_train_data"
eval_input_path: "data/test/tb_data/taobao_pdn_fake_test_data"
model_dir: "experiments/pdn_on_taobao"

train_config {
  log_step_count_steps: 50
  optimizer_config: {
    adam_optimizer: {
      learning_rate: {
        exponential_decay_learning_rate {
          initial_learning_rate: 0.001
          decay_steps: 1000
          decay_factor: 0.5
          min_learning_rate: 0.00001
        }
      }
    }
    use_moving_average: false
  }
  save_summary_steps: 50
  num_steps: 100
}

eval_config {
  metrics_set: {
    auc {}
  }
}

data_config {
  input_fields {
    input_name:'clk'
    input_type: INT32
  }
  input_fields {
    input_name:'buy'
    input_type: INT32
  }
  input_fields {
    input_name: 'pid'
    input_type: STRING
  }
  input_fields {
    input_name: 'adgroup_id'
    input_type: STRING
  }
  input_fields {
    input_name: 'cate_id'
    input_type: STRING
  }
  input_fields {
    input_name: 'campaign_id'
    input_type: STRING
  }
  input_fields {
    input_name: 'customer'
    input_type: STRING
  }
  input_fields {
    input_name: 'brand'
    input_type: STRING
  }
  input_fields {
    input_name: 'user_id'
    input_type: STRING
  }
  input_fields {
    input_name: 'cms_segid'
    input_type: STRING
  }
  input_fields {
    input_name: 'cms_group_id'
    input_type: STRING
  }
  input_fields {
    input_name: 'final_gender_code'
    input_type: STRING
  }
  input_fields {
    input_name: 'age_level'
    input_type: STRING
  }
  input_fields {
    input_name: 'pvalue_level'
    input_type: STRING
  }
  input_fields {
    input_name: 'shopping_level'
    input_type: STRING
  }
  input_fields {
    input_name: 'occupation'
    input_type: STRING
  }
  input_fields {
    input_name: 'new_user_class_level'
    input_type: STRING
  }
  input_fields {
    input_name: 'tag_category_list'
    input_type: STRING
  }
  input_fields {
    input_name: 'tag_brand_list'
    input_type: STRING
  }
  input_fields {
    input_name: 'price'
    input_type: INT32
  }
  input_fields {
    input_name: "i2i_rnk"
    input_type: STRING
  }
  input_fields {
    input_name: "event_type_seq"
    input_type: STRING
  }

  label_fields: 'clk'
  batch_size: 512
  num_epochs: 1
  prefetch_size: 32
  input_type: CSVInput
}

feature_configs : {
  input_names: 'pid'
  feature_type: IdFeature
  embedding_dim: 16
  hash_bucket_size: 10
}
feature_configs : {
  input_names: 'adgroup_id'
  feature_type: IdFeature
  embedding_dim: 16
  hash_bucket_size: 100000
}
feature_configs : {
  input_names: 'cate_id'
  feature_type: IdFeature
  embedding_dim: 16
  hash_bucket_size: 10000
}
feature_configs : {
  input_names: 'campaign_id'
  feature_type: IdFeature
  embedding_dim: 16
  hash_bucket_size: 100000
}
feature_configs : {
  input_names: 'customer'
  feature_type: IdFeature
  embedding_dim: 16
  hash_bucket_size: 100000
}
feature_configs : {
  input_names: 'brand'
  feature_type: IdFeature
  embedding_dim: 16
  hash_bucket_size: 100000
}
feature_configs : {
  input_names: 'user_id'
  feature_type: IdFeature
  embedding_dim: 16
  hash_bucket_size: 100000
}
feature_configs : {
  input_names: 'cms_segid'
  feature_type: IdFeature
  embedding_dim: 16
  hash_bucket_size: 100
}
feature_configs : {
  input_names: 'cms_group_id'
  feature_type: IdFeature
  embedding_dim: 16
  hash_bucket_size: 100
}
feature_configs : {
  input_names: 'final_gender_code'
  feature_type: IdFeature
  embedding_dim: 16
  hash_bucket_size: 10
}
feature_configs : {
  input_names: 'age_level'
  feature_type: IdFeature
  embedding_dim: 16
  hash_bucket_size: 10
}
feature_configs : {
  input_names: 'pvalue_level'
  feature_type: IdFeature
  embedding_dim: 16
  hash_bucket_size: 10
}
feature_configs : {
  input_names: 'shopping_level'
  feature_type: IdFeature
  embedding_dim: 16
  hash_bucket_size: 10
}
feature_configs : {
  input_names: 'occupation'
  feature_type: IdFeature
  embedding_dim: 16
  hash_bucket_size: 10
}
feature_configs : {
  input_names: 'new_user_class_level'
  feature_type: IdFeature
  embedding_dim: 16
  hash_bucket_size: 10
}
feature_configs : {
   input_names: 'tag_category_list'
   feature_type: SequenceFeature
   separator: '|'
   hash_bucket_size: 100000
   embedding_dim: 16
}
feature_configs : {
   input_names: 'event_type_seq'
   feature_type: SequenceFeature
   separator: '|'
   hash_bucket_size: 100000
   embedding_dim: 16
}
feature_configs : {
   input_names: 'tag_brand_list'
   feature_type: SequenceFeature
   separator: '|'
   hash_bucket_size: 100000
   embedding_dim: 16
}
feature_configs : {
  input_names: 'price'
  feature_type: IdFeature
  embedding_dim: 16
  num_buckets: 50
}
feature_configs {
  input_names: 'i2i_rnk'
  feature_type: SequenceFeature
  separator: '|'
  num_buckets: 128
  embedding_dim: 16
}

model_config:{
  model_class: "PDN"
  feature_groups: {
    group_name: 'u2i_seq'
    feature_names: 'event_type_seq'
  }
  feature_groups: {
    group_name: 'i_seq'
    feature_names: 'tag_category_list'
    feature_names: 'tag_brand_list'
  }
  feature_groups: {
    group_name: 'i2i_seq'
    feature_names: 'i2i_rnk'
  }
  feature_groups: {
    group_name: 'user'
    feature_names: 'user_id'
    feature_names: 'cms_segid'
    feature_names: 'cms_group_id'
    feature_names: 'age_level'
    feature_names: 'pvalue_level'
    feature_names: 'shopping_level'
    feature_names: 'occupation'
    feature_names: 'new_user_class_level'
    wide_deep:DEEP
  }
  feature_groups: {
    group_name: "item"
    feature_names: 'adgroup_id'
    feature_names: 'cate_id'
    feature_names: 'campaign_id'
    feature_names: 'customer'
    feature_names: 'brand'
    wide_deep:DEEP
  }
  feature_groups {
    group_name: 'bias'
    feature_names: 'user_id'
    feature_names: 'cms_segid'
    feature_names: 'cms_group_id'
    feature_names: 'age_level'
  }

  pdn {
    user_dnn {
      hidden_units: [128, 64, 32]
    }
    item_dnn {
      hidden_units: [128, 64, 32]
    }
    u2i_dnn {
      hidden_units: [64, 32]
    }
    trigger_dnn {
      hidden_units: [64, 32, 1]
    }
    i2i_dnn {
      hidden_units: [128, 64, 32]
    }
    sim_dnn {
      hidden_units: [64, 32, 1]
    }
    bias_dnn {
      hidden_units: [32, 32, 1]
    }

    l2_regularization: 1e-6
  }
  embedding_regularization: 5e-5
}

export_config {
}