模型调试

3d64bf0c · 宋柯 · bf69fbe1 · 3d64bf0c
Commit 3d64bf0c authored 3 years ago by 宋柯
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

featureEngSkTfRecord.py spark/featureEngSkTfRecord.py +2 -2

No files found.
--- a/spark/featureEngSkTfRecord.py
+++ b/spark/featureEngSkTfRecord.py
@@ -1116,11 +1116,11 @@ if __name__ == '__main__':

    output_file = "/data/files/wideAndDeep/train_samples.tfrecord"

-    train_samples = train_samples.rdd.map(get_example_string).collect()
+    train_samples_arr = train_samples.rdd.map(get_example_string).collect()
    train_samples.unpersist(True)

    writer = tf.io.TFRecordWriter(output_file)
-    for train_sample in train_samples:
+    for train_sample in train_samples_arr:
        writer.write(train_sample)
    writer.close()