Bläddra i källkod

Update category_models: fix add_constant

StrayWarrior 6 månader sedan
förälder
incheckning
e19d86d703
1 ändrade filer med 3 tillägg och 1 borttagningar
  1. 3 1
      src/long_articles/category_models.py

+ 3 - 1
src/long_articles/category_models.py

@@ -23,6 +23,8 @@ class CategoryRegressionV1:
             'CateOddities', 'CateFamily', 'CateHeartwarm',
             'CateHistory', 'CateHealth', 'CateLifeKnowledge', 'CateGossip',
             'CatePolitics', 'CateMilitary', 'CateMovie', 'CateSociety',
+            # 'CateFinTech', 'CateSocialCustom', 'CateWorkExperience',
+            # 'CateEducation',
             'view_count_rate' #, 'days_decrease'
         ]
 
@@ -75,7 +77,7 @@ class CategoryRegressionV1:
                                   print_p_value_threshold=0.1):
         X = df[self.features]  # 特征列
         y = df['RegressionY']  # 目标变量
-        X = sm.add_constant(X)
+        X = sm.add_constant(X, has_constant='add')
 
         model = sm.OLS(y, X).fit()