bug fix

cfddf34e · Your Name · 85f40df4 · cfddf34e
Commit cfddf34e authored Jun 06, 2019 by Your Name
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

dist_predict.py eda/esmm/Model_pipline/dist_predict.py +2 -2

No files found.
--- a/eda/esmm/Model_pipline/dist_predict.py
+++ b/eda/esmm/Model_pipline/dist_predict.py
@@ -187,8 +187,8 @@ if __name__ == "__main__":
    # df = spark.read.format("tfrecords").load(path+"nearby/part-r-00000")
    # df.show()

-    name = spark.createDataFrame([1,2,3,4,5],5)
-    test = name.map(lambda x: test_map())
+    name = spark.createDataFrame([1,2,3,4,5])
+    test = name.repartition(5).map(lambda x: test_map())
    test.show()

    b = time.time()