dbmtl_backbone_on_taobao.config

train_input_path: "data/test/tb_data/taobao_train_data"
eval_input_path: "data/test/tb_data/taobao_test_data"
model_dir: "experiments/dbmtl_backbone_taobao_ckpt"

train_config {
  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: 1e-07
        }
      }
    }
    use_moving_average: false
  }
  num_steps: 100
  sync_replicas: true
  save_checkpoints_steps: 100
  log_step_count_steps: 100
}
eval_config {
  metrics_set {
    auc {
    }
  }
}
data_config {
  batch_size: 4096
  label_fields: "clk"
  label_fields: "buy"
  prefetch_size: 32
  input_type: CSVInput
  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
  }
}
feature_config: {
  features {
    input_names: "pid"
    feature_type: IdFeature
    embedding_dim: 16
    hash_bucket_size: 10
  }
  features {
    input_names: "adgroup_id"
    feature_type: IdFeature
    embedding_dim: 16
    hash_bucket_size: 100000
  }
  features {
    input_names: "cate_id"
    feature_type: IdFeature
    embedding_dim: 16
    hash_bucket_size: 10000
  }
  features {
    input_names: "campaign_id"
    feature_type: IdFeature
    embedding_dim: 16
    hash_bucket_size: 100000
  }
  features {
    input_names: "customer"
    feature_type: IdFeature
    embedding_dim: 16
    hash_bucket_size: 100000
  }
  features {
    input_names: "brand"
    feature_type: IdFeature
    embedding_dim: 16
    hash_bucket_size: 100000
  }
  features {
    input_names: "user_id"
    feature_type: IdFeature
    embedding_dim: 16
    hash_bucket_size: 100000
  }
  features {
    input_names: "cms_segid"
    feature_type: IdFeature
    embedding_dim: 16
    hash_bucket_size: 100
  }
  features {
    input_names: "cms_group_id"
    feature_type: IdFeature
    embedding_dim: 16
    hash_bucket_size: 100
  }
  features {
    input_names: "final_gender_code"
    feature_type: IdFeature
    embedding_dim: 16
    hash_bucket_size: 10
  }
  features {
    input_names: "age_level"
    feature_type: IdFeature
    embedding_dim: 16
    hash_bucket_size: 10
  }
  features {
    input_names: "pvalue_level"
    feature_type: IdFeature
    embedding_dim: 16
    hash_bucket_size: 10
  }
  features {
    input_names: "shopping_level"
    feature_type: IdFeature
    embedding_dim: 16
    hash_bucket_size: 10
  }
  features {
    input_names: "occupation"
    feature_type: IdFeature
    embedding_dim: 16
    hash_bucket_size: 10
  }
  features {
    input_names: "new_user_class_level"
    feature_type: IdFeature
    embedding_dim: 16
    hash_bucket_size: 10
  }
  features {
    input_names: "tag_category_list"
    feature_type: TagFeature
    embedding_dim: 16
    hash_bucket_size: 100000
    separator: "|"
  }
  features {
    input_names: "tag_brand_list"
    feature_type: TagFeature
    embedding_dim: 16
    hash_bucket_size: 100000
    separator: "|"
  }
  features {
    input_names: "price"
    feature_type: IdFeature
    embedding_dim: 16
    num_buckets: 50
  }
}
model_config {
  model_name: "DBMTL"
  model_class: "MultiTaskModel"
  feature_groups {
    group_name: "all"
    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"
    feature_names: "adgroup_id"
    feature_names: "cate_id"
    feature_names: "campaign_id"
    feature_names: "customer"
    feature_names: "brand"
    feature_names: "price"
    feature_names: "pid"
    feature_names: "tag_category_list"
    feature_names: "tag_brand_list"
    wide_deep: DEEP
  }
  backbone {
    blocks {
      name: "mask_net"
      inputs {
        feature_group_name: "all"
      }
      keras_layer {
        class_name: 'MaskNet'
        masknet {
          mask_blocks {
            aggregation_size: 512
            output_size: 256
          }
          mask_blocks {
            aggregation_size: 512
            output_size: 256
          }
          mask_blocks {
            aggregation_size: 512
            output_size: 256
          }
          mlp {
            hidden_units: [512, 256]
          }
        }
      }
    }
  }
  model_params {
    task_towers {
      tower_name: "ctr"
      label_name: "clk"
      loss_type: CLASSIFICATION
      metrics_set: {
        auc {}
      }
      dnn {
        hidden_units: [256, 128, 64, 32]
      }
      relation_dnn {
        hidden_units: [32]
      }
      weight: 1.0
    }
    task_towers {
      tower_name: "cvr"
      label_name: "buy"
      loss_type: CLASSIFICATION
      metrics_set: {
        auc {}
      }
      dnn {
        hidden_units: [256, 128, 64, 32]
      }
      relation_tower_names: ["ctr"]
      relation_dnn {
        hidden_units: [32]
      }
      weight: 1.0
    }
    l2_regularization: 1e-6
  }
  embedding_regularization: 5e-6
}