Commit 4ddd65e7 authored by 张彦钊's avatar 张彦钊

change sql for test

parent 7795f770
......@@ -236,7 +236,8 @@ 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")
predict_list.extend(["B2F0665E-4375-4169-8FE3-8A26A1CFE248","AB20292B-5D15-4C44-9429-1C2FF5ED26F6","358035085192742"])
# 只预测尾号是6的ID,这块也可以在数据库取数据时过滤一下
# predict_list = list(filter(lambda x:str(x)[-1] == "6", predict_list))
start = time.time()
......@@ -252,7 +253,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("B2F0665E-4375-4169-8FE3-8A26A1CFE248" in predict_list)
print(predict_list)
print(end - start)
......
......@@ -7,7 +7,7 @@ def get_active_users():
now = datetime.now()
now_start = str(now)[:16] + ":00"
now_end = str(now)[:16] + ":59"
sql = "select device_id from user_active_time order by active_time desc limit 100;"
sql = "select device_id from user_active_time order by active_time desc limit 1;"
# sql = "select device_id from user_active_time " \
# "where active_time <= '{}' and active_time >= '{}'".format(now_end,now_start)
device_id_df = con_sql(sql)
......
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