Commit 52960316 authored by 张彦钊's avatar 张彦钊

change sql for test

parent 9882d58f
...@@ -239,7 +239,7 @@ if __name__ == "__main__": ...@@ -239,7 +239,7 @@ if __name__ == "__main__":
# 求活跃用户和老用户的交集,也就是只预测老用户 # 求活跃用户和老用户的交集,也就是只预测老用户
predict_list = list(set(device_id_list) & set(old_device_id_list)) predict_list = list(set(device_id_list) & set(old_device_id_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()
warnings.filterwarnings("ignore") warnings.filterwarnings("ignore")
data_set_cid = pd.read_csv(DIRECTORY_PATH + "data_set_cid.csv")["cid"].values.tolist() data_set_cid = pd.read_csv(DIRECTORY_PATH + "data_set_cid.csv")["cid"].values.tolist()
......
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