Commit 7795f770 authored by 张彦钊's avatar 张彦钊

change sql for test

parent 4bfec783
......@@ -236,7 +236,7 @@ if __name__ == "__main__":
old_device_id_list = pd.read_csv(DIRECTORY_PATH + "data_set_device_id.csv")["device_id"].values.tolist()
# 求活跃用户和老用户的交集,也就是只预测老用户
predict_list = list(set(device_id_list) & set(old_device_id_list))
predict_list.append("65EC6C14-1AD6-44C2-AED2-C41452284E91")
# 只预测尾号是6的ID,这块也可以在数据库取数据时过滤一下
# predict_list = list(filter(lambda x:str(x)[-1] == "6", predict_list))
start = time.time()
......@@ -252,6 +252,7 @@ if __name__ == "__main__":
print("在不在")
print("358035085192742" in predict_list)
print("AB20292B-5D15-4C44-9429-1C2FF5ED26F6" in predict_list)
print("65EC6C14-1AD6-44C2-AED2-C41452284E91" in predict_list)
print(predict_list)
print(end - start)
......
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