Commit 50d76988 authored by 张彦钊's avatar 张彦钊

add print

parent 952d3933
......@@ -217,22 +217,30 @@ def multi_update(key, name_dict):
if __name__ == "__main__":
start = time.time()
warnings.filterwarnings("ignore")
# TODO 上线后把预测用户改成多进程预测
data_set_cid = pd.read_csv(DIRECTORY_PATH + "data_set_cid.csv")["cid"].values.tolist()
device_id = "358035085192742"
native_queue_list, nearby_queue_list, nation_queue_list, megacity_queue_list = test_con_sql(device_id)
name_dict = {"native_queue": native_queue_list, "nearby_queue": nearby_queue_list,
"nation_queue": nation_queue_list, "megacity_queue": megacity_queue_list}
pool = Pool(4)
for key in name_dict.keys():
multi_update(key, name_dict)
pool.apply_async(multi_update,(key,name_dict,))
pool.close()
pool.join()
# # TODO 上线后把预测用户改成多进程预测
# data_set_cid = pd.read_csv(DIRECTORY_PATH + "data_set_cid.csv")["cid"].values.tolist()
#
# device_id = "358035085192742"
# native_queue_list, nearby_queue_list, nation_queue_list, megacity_queue_list = test_con_sql(device_id)
# name_dict = {"native_queue": native_queue_list, "nearby_queue": nearby_queue_list,
# "nation_queue": nation_queue_list, "megacity_queue": megacity_queue_list}
#
# for key in name_dict.keys():
# multi_update(key, name_dict)
end = time.time()
print(end-start)
# pool = Pool(4)
# for key in name_dict.keys():
# pool.apply_async(multi_update,(key,name_dict,))
# pool.close()
# pool.join()
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