Explorar o código

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

luojunhui hai 5 meses
pai
achega
c9c36aaef5
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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