build_model.py 文件源码

def feature_selection_logit(xtr, ytr):
    model = LogisticRegressionCV(penalty='l1', solver='liblinear', cv=5)
    model.fit(xtr, ytr)
    columns = np.arange(xtr.shape[1])[~np.isclose(model.coef_.ravel(), 0)]
    return columns