Commit 67ef1075 authored by Your Name's avatar Your Name

predict add sample id

parent 61225594
...@@ -210,7 +210,8 @@ if __name__ == "__main__": ...@@ -210,7 +210,8 @@ if __name__ == "__main__":
print(rdd_te_files.collect()) print(rdd_te_files.collect())
print("-" * 100) print("-" * 100)
indices = rdd_te_files.repartition(100).map(lambda x: main(x)) indices = rdd_te_files.repartition(100).map(lambda x: main(x))
print(indices.take(1)) print(indices.take(2))
print("耗时(秒):") print("耗时(秒):")
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment