Skip to content

F
ffm-baseline
Commit f8fd7e55 authored by 张彦钊's avatar 张彦钊
Browse files
Options

change test file

parent 0aae1e2c
Show whitespace changes
Inline Side-by-side
Showing with 17 additions and 22 deletions
eda/esmm/Model_pipline/train.py
View file @ f8fd7e55
......@@ -298,28 +298,23 @@ def main(_):
FLAGS.model_dir = FLAGS.model_dir + FLAGS.dt_dir
#FLAGS.data_dir = FLAGS.data_dir + FLAGS.dt_dir
print('task_type ', FLAGS.task_type)
print('model_dir ', FLAGS.model_dir)
print('hdfs_dir ', FLAGS.hdfs_dir)
print('local_dir ', FLAGS.local_dir)
print('dt_dir ', FLAGS.dt_dir)
print('num_epochs ', FLAGS.num_epochs)
print('feature_size ', FLAGS.feature_size)
print('field_size ', FLAGS.field_size)
print('embedding_size ', FLAGS.embedding_size)
print('batch_size ', FLAGS.batch_size)
print('deep_layers ', FLAGS.deep_layers)
print('dropout ', FLAGS.dropout)
print('loss_type ', FLAGS.loss_type)
print('optimizer ', FLAGS.optimizer)
print('learning_rate ', FLAGS.learning_rate)
print('l2_reg ', FLAGS.l2_reg)
print('ctr_task_wgt ', FLAGS.ctr_task_wgt)
#------init Envs------
# tr_files = [path+"tr/part-r-00000"]
# va_files = [path+"va/part-r-00000"]
# te_files = ["%s/part-r-00000" % FLAGS.hdfs_dir]
# print('task_type ', FLAGS.task_type)
# print('model_dir ', FLAGS.model_dir)
# print('hdfs_dir ', FLAGS.hdfs_dir)
# print('local_dir ', FLAGS.local_dir)
# print('dt_dir ', FLAGS.dt_dir)
# print('num_epochs ', FLAGS.num_epochs)
# print('feature_size ', FLAGS.feature_size)
# print('field_size ', FLAGS.field_size)
# print('embedding_size ', FLAGS.embedding_size)
# print('batch_size ', FLAGS.batch_size)
# print('deep_layers ', FLAGS.deep_layers)
# print('dropout ', FLAGS.dropout)
# print('loss_type ', FLAGS.loss_type)
# print('optimizer ', FLAGS.optimizer)
# print('learning_rate ', FLAGS.learning_rate)
# print('l2_reg ', FLAGS.l2_reg)
# print('ctr_task_wgt ', FLAGS.ctr_task_wgt)
tr_files = get_filename("/strategy/esmm/tr")
va_files = get_filename("/strategy/esmm/va")
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment