模型调试

spark/featureEngSkTfRecordTest.py

@@ -940,7 +940,7 @@ if __name__ == '__main__':
     # train_samples.write.mode("overwrite").options(header="false", sep='|').csv(output_file)
 
     import tensorflow as tf
-    def get_example_string(line):
+    def get_example_string(splits):
         features = {
             'ITEM_CATEGORY_card_id': tf.train.Feature(bytes_list=tf.train.BytesList(value=[splits[0].encode()])),
             'USER_CATEGORY_device_id': tf.train.Feature(bytes_list=tf.train.BytesList(value=[splits[2].encode()])),