Commit 1ffd2520 authored by 张彦钊's avatar 张彦钊

修改rerank文件每次写入的数量

parent 2daf15e5
......@@ -204,6 +204,9 @@ if __name__ == "__main__":
df = df.rename(columns={0: "device_id", 1: "city_id",2:"native_queue",
3:"nearby_queue",4:"nation_queue",5:"megacity_queue",6:"time"})
print("df numbers")
print(df.shape[0])
to_data_base(df)
......
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