修改字典生成方式

tensnsorflow/pipe.sh

@@ -43,12 +43,4 @@ ${PYTHON_PATH} ${MODEL_PATH}/train.py --ctr_task_wgt=0.3 --learning_rate=0.0001
 echo "infer nearby..."
 ${PYTHON_PATH} ${MODEL_PATH}/train.py --ctr_task_wgt=0.3 --learning_rate=0.0001 --deep_layers=256,128 --dropout=0.8,0.5 --optimizer=Adam --num_epochs=1 --embedding_size=16 --batch_size=1024 --field_size=10 --feature_size=2000 --l2_reg=0.005 --log_steps=100 --num_threads=36 --model_dir=${DATA_PATH}/model_ckpt/DeepCvrMTL/ --data_dir=${DATA_PATH}/nearby --task_type=infer > ${DATA_PATH}/infer.log
 
-echo "sort and 2sql"
-${PYTHON_PATH} ${OLD_PATH}/Model_pipline/sort_and_2sql.py
-
-echo "infer and sort and 2sql time"
-current=$(date "+%Y-%m-%d %H:%M:%S")
-timeStamp=$(date -d "$current" +%s)
-currentTimeStamp=$((timeStamp*1000+`date "+%N"`/1000000))
-echo $current
 

tensnsorflow/test.py

@@ -29,8 +29,8 @@ def gen_tfrecords(in_file):
 
     for i in range(df.shape[0]):
         features = tf.train.Features(feature={
-            "y": tf.train.Feature(int64_list=tf.train.Int64List(value=[df["y"][i]])),
-            "z": tf.train.Feature(int64_list=tf.train.Int64List(value=[df["z"][i]])),
+            "y": tf.train.Feature(float_list=tf.train.FloatList(value=[df["y"][i]])),
+            "z": tf.train.Feature(float_list=tf.train.FloatList(value=[df["z"][i]])),
             "top": tf.train.Feature(int64_list=tf.train.Int64List(value=[df["top"][i]])),
             "channel": tf.train.Feature(int64_list=tf.train.Int64List(value=[df["channel"][i]])),
             "ucity_id": tf.train.Feature(int64_list=tf.train.Int64List(value=[df["ucity_id"][i]])),

tensnsorflow/train.py

@@ -51,8 +51,8 @@ def input_fn(filenames, batch_size=32, num_epochs=1, perform_shuffle=False):
     print('Parsing', filenames)
     def _parse_fn(record):
         features = {
-            "y": tf.FixedLenFeature([], tf.int64),
-            "z": tf.FixedLenFeature([], tf.int64),
+            "y": tf.FixedLenFeature([], tf.float32),
+            "z": tf.FixedLenFeature([], tf.float32),
             "top": tf.FixedLenFeature([], tf.int64),
             "channel": tf.FixedLenFeature([], tf.int64),
             "ucity_id": tf.FixedLenFeature([], tf.int64),