|
@@ -23,6 +23,8 @@ class CategoryRegressionV1:
|
|
|
'CateOddities', 'CateFamily', 'CateHeartwarm',
|
|
|
'CateHistory', 'CateHealth', 'CateLifeKnowledge', 'CateGossip',
|
|
|
'CatePolitics', 'CateMilitary', 'CateMovie', 'CateSociety',
|
|
|
+
|
|
|
+
|
|
|
'view_count_rate'
|
|
|
]
|
|
|
|
|
@@ -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()
|
|
|
|