change test

rerank_esmm.py

@@ -182,25 +182,28 @@ if __name__ == "__main__":
     print(len(users_list))
     total_samples = list()
     name_tag = get_searchworlds_to_tagid()
-    for uid_city in users_list:
-        tag_list = get_user_profile(uid_city[0])
-        queues = get_queues(uid_city[0], uid_city[1])
-        if len(queues) > 0 and len(tag_list) > 0:
-            new_native = tag_boost(queues[0], tag_list)
-            new_nearby = tag_boost(queues[1], tag_list)
-
-            insert_time = str(datetime.datetime.now().strftime('%Y%m%d%H%M'))
-            sample = [uid_city[0], uid_city[1], new_native, new_nearby, queues[2], queues[3], insert_time]
-            total_samples.append(sample)
-            print(len(total_samples))
-
-    if len(total_samples) > 0:
-        df = pd.DataFrame(total_samples)
-        df = df.rename(columns={0: "device_id", 1: "city_id",2:"native_queue",
-                                3:"nearby_queue",4:"nation_queue",5:"megacity_queue",6:"time"})
-
-        to_data_base(df)
-        print("good boy")
+    n = 600
+    users_list = [a[i:i + n] for i in range(0, len(users_list), n)]
+    for child_users_list in users_list:
+        for uid_city in child_users_list:
+            tag_list = get_user_profile(uid_city[0])
+            queues = get_queues(uid_city[0], uid_city[1])
+            if len(queues) > 0 and len(tag_list) > 0:
+                new_native = tag_boost(queues[0], tag_list)
+                new_nearby = tag_boost(queues[1], tag_list)
+
+                insert_time = str(datetime.datetime.now().strftime('%Y%m%d%H%M'))
+                sample = [uid_city[0], uid_city[1], new_native, new_nearby, queues[2], queues[3], insert_time]
+                total_samples.append(sample)
+                print(len(total_samples))
+
+        if len(total_samples) > 0:
+            df = pd.DataFrame(total_samples)
+            df = df.rename(columns={0: "device_id", 1: "city_id",2:"native_queue",
+                                    3:"nearby_queue",4:"nation_queue",5:"megacity_queue",6:"time"})
+
+            to_data_base(df)
+            print("good boy")
 
     # TODO to kv