Commit 51fbf748 authored by 张彦钊's avatar 张彦钊

change sql for test

parent 66098682
...@@ -236,10 +236,7 @@ if __name__ == "__main__": ...@@ -236,10 +236,7 @@ if __name__ == "__main__":
old_device_id_list = pd.read_csv(DIRECTORY_PATH + "data_set_device_id.csv")["device_id"].values.tolist() 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 = list(set(device_id_list) & set(old_device_id_list))
print("在不在")
print("358035085192742" in predict_list)
print("AB20292B-5D15-4C44-9429-1C2FF5ED26F6" in predict_list)
print(predict_list)
# 只预测尾号是6的ID,这块也可以在数据库取数据时过滤一下 # 只预测尾号是6的ID,这块也可以在数据库取数据时过滤一下
# predict_list = list(filter(lambda x:str(x)[-1] == "6", predict_list)) # predict_list = list(filter(lambda x:str(x)[-1] == "6", predict_list))
start = time.time() start = time.time()
...@@ -252,6 +249,10 @@ if __name__ == "__main__": ...@@ -252,6 +249,10 @@ if __name__ == "__main__":
else: else:
print("该列表是新用户,不需要预测") print("该列表是新用户,不需要预测")
end = time.time() end = time.time()
print("在不在")
print("358035085192742" in predict_list)
print("AB20292B-5D15-4C44-9429-1C2FF5ED26F6" in predict_list)
print(predict_list)
print(end - start) 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