scikit-lego icon indicating copy to clipboard operation
scikit-lego copied to clipboard

[FEATURE] equal_opportunity_score positive_target should match pos_label to adhere to sklearn api

Open MBrouns opened this issue 5 years ago • 0 comments

from sklearn.model_selection import cross_validate, StratifiedKFold
from sklearn.metrics import precision_score, recall_score, make_scorer
cross_validate(
    pipeline, X, y,
    scoring = {
        'eq_op_colour': equal_opportunity_score('colour', positive_target='Yes'),
        'eq_op_age': equal_opportunity_score('age', positive_target='Yes'),
        'eq_op_sex': equal_opportunity_score('sex', positive_target='Yes'),
        'precision': make_scorer(precision_score, pos_label='Yes'),
        'recall': make_scorer(recall_score, pos_label='Yes')
    },
    cv=StratifiedKFold(5)
)

MBrouns avatar Jan 15 '20 09:01 MBrouns