inFairness icon indicating copy to clipboard operation
inFairness copied to clipboard

add support to fetch trained logistic regression model in the logreg metric

Open MayankAgarwal opened this issue 1 year ago • 0 comments

fixes #58

Adds a property to get trained Logistic regression models for a fitted Logistic Regression Sensitive Subspace metric.

metric = inFairness.distances.LogisticRegSensitiveSubspace()
models = metric.logistic_regression_models

The models variable will now contain a list of logistic regression models modeling P(K|X)

Note: cannot add this to the state of the metric since state_dict only supports tensors to be specified, whereas this is a sklearn object.

MayankAgarwal avatar Oct 01 '22 00:10 MayankAgarwal