Merge pull request #380 from frankwhzhang/fix_esmm_0208

fix esmm

Author: frankwhzhang

datasets/ali-ccp/run.sh

@@ -1,5 +1,7 @@
 mkdir train_data
 mkdir test_data
 
-wget -P train_data/ https://paddlerec.bj.bcebos.com/esmm/traindata.csv
-wget -P test_data/ https://paddlerec.bj.bcebos.com/esmm/testdata.csv
+wget https://paddlerec.bj.bcebos.com/esmm/traindata_10w.csv  
+wget https://paddlerec.bj.bcebos.com/esmm/testdata_10w.csv 
+mv traindata_10w.csv train_data
+mv testdata_10w.csv test_data

models/multitask/esmm/config_bigdata.yaml

@@ -20,7 +20,7 @@ runner:
   use_auc: True
   train_batch_size: 1024
   epochs: 10
-  print_interval: 100
+  print_interval: 10
   #model_init_path: "output_model/0" # init model
   model_save_path: "output_model_esmm_all"
   test_data_dir: "../../../datasets/ali-ccp/test_data"

tools/trainer.py

@@ -123,7 +123,7 @@ def main(args):
                 for metric_id in range(len(metric_list_name)):
                     metric_str += (
                         metric_list_name[metric_id] +
-                        ": {:.6f},".format(metric_list[metric_id].accumulate())
+                        ":{:.6f}, ".format(metric_list[metric_id].accumulate())
                     )
                 tensor_print_str = ""
                 if tensor_print_dict is not None: