修改测试文件

tensnsorflow/record.py

@@ -35,7 +35,7 @@ def gen_tfrecords(in_file):
         for j in feats:
             id = np.append(id,df[j][i])
         app_list = np.array(str(df["app_list"][i]).split(","))
-        level2_list = np.array(str(df["clevel2_id"][i]).split(","))
+        level2_list = np.array(str(df["level2_ids"][i]).split(","))
         level3_list = np.array(str(df["level3_ids"][i]).split(","))
         features = tf.train.Features(feature={
             "y": tf.train.Feature(float_list=tf.train.FloatList(value=[df["y"][i]])),