Commit 5da4b9b8 authored by 张彦钊's avatar 张彦钊

change sql for test

parent 73f0d91f
......@@ -240,13 +240,14 @@ 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("AB20292B-5D15-4C44-9429-1C2FF5ED26F6")
predict_list.append("358035085192742")
# 只预测尾号是6的ID,这块也可以在数据库取数据时过滤一下
# predict_list = list(filter(lambda x:str(x)[-1] == "6", predict_list))
start = time.time()
warnings.filterwarnings("ignore")
data_set_cid = pd.read_csv(DIRECTORY_PATH + "data_set_cid.csv")["cid"].values.tolist()
print("predictlist")
print(predict_list[:6])
if predict_list != []:
for device_id in predict_list:
user_update(device_id)
......
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