Skip to content

Commit 2743bde

Browse files
committed
fix esmm
1 parent 6f7dcd3 commit 2743bde

File tree

3 files changed

+6
-4
lines changed

3 files changed

+6
-4
lines changed

datasets/ali-ccp/run.sh

+4-2
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
mkdir train_data
22
mkdir test_data
33

4-
wget -P train_data/ https://paddlerec.bj.bcebos.com/esmm/traindata.csv
5-
wget -P test_data/ https://paddlerec.bj.bcebos.com/esmm/testdata.csv
4+
wget https://paddlerec.bj.bcebos.com/esmm/traindata_10w.csv
5+
wget https://paddlerec.bj.bcebos.com/esmm/testdata_10w.csv
6+
mv traindata_10w.csv train_data
7+
mv testdata_10w.csv test_data

models/multitask/esmm/config_bigdata.yaml

+1-1
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ runner:
2020
use_auc: True
2121
train_batch_size: 1024
2222
epochs: 10
23-
print_interval: 100
23+
print_interval: 10
2424
#model_init_path: "output_model/0" # init model
2525
model_save_path: "output_model_esmm_all"
2626
test_data_dir: "../../../datasets/ali-ccp/test_data"

tools/trainer.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def main(args):
123123
for metric_id in range(len(metric_list_name)):
124124
metric_str += (
125125
metric_list_name[metric_id] +
126-
": {:.6f},".format(metric_list[metric_id].accumulate())
126+
":{:.6f}, ".format(metric_list[metric_id].accumulate())
127127
)
128128
tensor_print_str = ""
129129
if tensor_print_dict is not None:

0 commit comments

Comments
 (0)