Jelajahi Sumber

Update run_category_model_v1: change negative cate clip

StrayWarrior 4 bulan lalu
induk
melakukan
883522d887
1 mengubah file dengan 3 tambahan dan 1 penghapusan
  1. 3 1
      run_category_model_v1.py

+ 3 - 1
run_category_model_v1.py

@@ -144,7 +144,7 @@ def main():
             if abs(param) > 0.1 and p_value < P_VALUE_THRESHOLD and cate_name is not None:
                 scale_factor = min(0.1 / p_value, 1)
                 print(f"{account_id} {account_name} {cate_name} {param:.3f} {p_value:.3f}")
-                truncate_param = round(max(min(param, 0.25), -0.3) * scale_factor, 6)
+                truncate_param = round(max(min(param, 0.25), -0.25) * scale_factor, 6)
                 current_record['category_map'][cate_name] = truncate_param
             # 用于冷启文章分配的负向品类
             if param < -0.1 and cate_name is not None and p_value < P_VALUE_THRESHOLD:
@@ -185,4 +185,6 @@ def main():
     # print(json.dumps(account_negative_cates, ensure_ascii=False, indent=2))
 
 if __name__ == '__main__':
+    pd.set_option('display.max_columns', None)
+    pd.set_option('display.max_rows', None)
     main()