Browse Source

冷启动发布,增加品类信息bugfix

luojunhui 1 tháng trước cách đây
mục cha
commit
c9c36aaef5
1 tập tin đã thay đổi với 1 bổ sung1 xóa
  1. 1 1
      cold_start/publish/publishCategoryArticles.py

+ 1 - 1
cold_start/publish/publishCategoryArticles.py

@@ -344,7 +344,7 @@ class CategoryColdStartTask(object):
 
         # split into different category
         for ai_category in self.article_category_list:
-            filter_category_df = filtered_articles_df[filtered_articles_df['ai_category'] == ai_category]
+            filter_category_df = filtered_articles_df[filtered_articles_df['category_by_ai'] == ai_category]
             url_list = filter_category_df['link'].values.tolist()
             if url_list:
                 # create_crawler_plan