Browse Source

Update accountServer: optimize sim type function

StrayWarrior 8 months ago
parent
commit
639923d63b
1 changed files with 10 additions and 7 deletions
  1. 10 7
      routes/accountServer.py

+ 10 - 7
routes/accountServer.py

@@ -102,14 +102,14 @@ class AccountServer(object):
             rate=rate,
             view_count_filter=view_count_filter,
         )
-        if interest_type == "account_avg":
-            interest_weight = (good_df["show_view_count"] / good_df["view_count_avg"]).values.tolist()
-        else:
-            interest_weight = good_df["show_view_count"].values.tolist()
+        extend_dicts = {
+            'view_count': good_df["show_view_count"].values.tolist(),
+            'view_count_rate': (good_df["show_view_count"] / good_df["view_count_avg"]).values.tolist()
+        }
         account_interest = good_df["title"].values.tolist()
         print(account_interest)
-        print(interest_weight)
-        return account_interest, interest_weight
+        print(extend_dicts)
+        return account_interest, extend_dicts
 
     async def get_each_account_score_list(self, account_name):
         """
@@ -117,7 +117,7 @@ class AccountServer(object):
         :return:
         """
         try:
-            account_interest, interest_weight = await self.get_account_interest(
+            account_interest, extend_dicts = await self.get_account_interest(
                 account_name=account_name,
                 interest_type=self.interest_type,
                 rate=self.rate,
@@ -125,6 +125,9 @@ class AccountServer(object):
                 min_time=self.min_time,
                 max_time=self.max_time,
             )
+            interest_weight = extend_dicts['view_count']
+            if self.sim_type == "weighted_by_view_count_rate":
+                extend_dicts['view_count_rate']
             response = await self.request_for_nlp(
                 title_list=self.title_list,
                 account_interest=account_interest,