bug fix

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

Showing with 1 addition and 1 deletion

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

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

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