def test_drop_vif(): df = drop_vif(frame.fillna(-1), exclude='target') assert ['C', 'F', 'target'] == df.columns.tolist()
def test_stepwise_return_drop(): df, drop_list = stepwise(frame.fillna(-1), target='target', return_drop=True) assert ['B', 'A', 'D'] == drop_list
def test_stepwise_exclude(): df = stepwise(frame.fillna(-1), target='target', exclude='B') assert ['B', 'C', 'E', 'F', 'target'] == df.columns.tolist()
def test_stepwise_forward(): df = stepwise(frame.fillna(-1), target='target', direction='forward') assert ['C', 'E', 'F', 'target'] == df.columns.tolist()
def test_stepwise_ks(): df = stepwise(frame.fillna(-1), target = 'target', criterion = 'ks', direction = 'forward') assert ['A', 'C', 'target'] == df.columns.tolist()
def test_stepwise_lr(): df = stepwise(frame.fillna(-1), target = 'target', estimator = 'lr', direction = 'forward') assert ['C', 'target'] == df.columns.tolist()