facebookresearch/balance 0.8.0

0.8.0 (2023-04-26)

on GitHub

New Features

• Add rake method to .adjust (currently in beta, given that it doesn't handles marginal target as input).
• Add a new function prepare_marginal_dist_for_raking - to take in a dict of marginal proportions and turn them into a pandas DataFrame. This can serve as an input target population for raking.

Misc

• The ipw function now gets max_de=None as default (instead of 1.5). This version is faster, and the user can still choose a threshold as desired.
• Adding hex stickers graphics files

Documentation

• New section on raking.
• New notebook (in the tutorial section):
  • quickstart_rake - like the quickstart tutorial, but shows how to use the rake (raking) algorithm and compares the results to IPW (logistic regression with LASSO).

Contributors

@talgalili, @SarigT