Merge pull request #718 from yinhaofeng/dcn

dcn_config

Author: frankwhzhang

models/rank/dcn/README.md

@@ -106,7 +106,7 @@ deepAndCross模型的组网本质是一个二分类任务，模型代码参考
 
 | 模型 | auc | batch_size | epoch_num| Time of each epoch |
 | :------| :------ | :------ | :------| :------ | 
-| dcn |  0.777   | 32  |  10  | 约 3 小时 |
+| dcn |  0.777   | 512  |  10  | 约 3 小时 |
 
 1. 确认您当前所在目录为PaddleRec/models/rank/dcn
 2. 在"criteo data"全量数据目录下，运行数据一键处理脚本，命令如下：  

models/rank/dcn/config_bigdata.yaml

@@ -0,0 +1,56 @@
+# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+runner:
+  train_data_dir: "../../../datasets/criteo/slot_train_data_full"
+  train_reader_path: "reader" # importlib format
+  use_gpu: True
+  use_auc: True
+  train_batch_size: 512
+  epochs: 10
+  print_interval: 10
+  #model_init_path: "output_model/0" # init model
+  model_save_path: "output_model_dcn_all"
+  test_data_dir: "../../../datasets/criteo/slot_test_data_full"
+  infer_reader_path: "reader" # importlib format
+  infer_batch_size: 512
+  infer_load_path: "output_model_dcn_all"
+  infer_start_epoch: 0
+  infer_end_epoch: 10
+
+
+# hyper parameters of user-defined network
+hyper_parameters:
+  # optimizer config
+  optimizer:
+    class: Adam
+    learning_rate: 0.0001
+    strategy: async
+  # user-defined <key, value> pairs
+  sparse_inputs_slots: 27
+  sparse_feature_number: 1000001
+  sparse_feature_dim: 9
+  dense_input_dim: 13
+  fc_sizes: [512, 256, 128] #, 32]
+  distributed_embedding: 0
+
+# sparse_inputs_slots + dense_input_dim
+
+  cross_num: 2
+  l2_reg_cross: 0.00005
+  dnn_use_bn: False
+  clip_by_norm: 100.0
+  is_sparse: False
+  # cat_feat_num: "{workspace}/data/sample_data/cat_feature_num.txt"