search

microsoft / responsible-ai-toolbox-mitigations

Python library for implementing Responsible AI mitigations.
https://responsible-ai-toolbox-mitigations.readthedocs.io/en/latest/
MIT License
57 stars 6 forks source link

Balancing on multiple columns work around #13

Closed akshara-msft closed 1 year ago

akshara-msft commented 2 years ago

Currently there is a work around in the end to end Jupyter notebook so that we can balance on two columns at the same time using any of the rebalancing techniques. While making API deseign changes to the Rebalance API, we should allow the user to specific multiple columns that they are interested in balancing on rather than creating a single column and balancing on that single column cohort

mrfmendonca commented 1 year ago

This has been fixed in PR #45, which introduces the CohortManager class. This class allows you to rebalance each cohort separately, without requiring any workarounds. More details can be found in the following notebook: /notebooks/cohort/cohort_manager.ipynb.