Skip to content

Commit 9fbbf2c

Browse files
committed
intercept added
1 parent 6122984 commit 9fbbf2c

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

cobra/model_selection.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -234,8 +234,8 @@ def _forwardSelection(self, df_sel, forced_vars, excluded_vars):
234234
predictors.append(add_variable)
235235

236236
#Return only DF with selected models
237-
clmns_out = ['step', 'coef', 'auc_train', 'auc_selection', 'auc_validation', 'predictors_subset', 'last_var_added',
238-
'pred_training','pred_selection','pred_validation']
237+
clmns_out = ['step', 'coef', 'intercept', 'auc_train', 'auc_selection', 'auc_validation',
238+
'predictors_subset', 'last_var_added','pred_training','pred_selection','pred_validation']
239239

240240
df_out = df_forward_selection[clmns_out][df_forward_selection['selected_model'] == True]
241241

0 commit comments

Comments
 (0)