Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add maxiter and maxfun optimisation arguments to LFR fit method #184

Merged

Conversation

tschwedes
Copy link
Contributor

The fit method for the LFR class has so far pre-defined arguments for the L-BFGS-B optimisation method, which the user cannot change. We hereby emphasise maxiter and maxfun, denoting the maximum number of iterations and function evaluations, which are pre-set to a value of 5000.

In some experiments, these values can be too small to allow for convergence of the optimisation method. This can lead to suboptimal solutions, hence suboptimal results for accuracy and fairness in predictions based on the learnt fair representation.

We added the arguments maxiter and maxfun to the fit method, which are set by default to the original values. However the user has now the freedom to change those values if required to ensure convergence.

Copy link

@psattige psattige left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for the PR. Looks good to me.

@hoffmansc @nrkarthikeyan Any thoughts?

@nrkarthikeyan nrkarthikeyan merged commit 4128017 into Trusted-AI:master Jun 17, 2020
Illia-Kryvoviaz pushed a commit to Illia-Kryvoviaz/AIF360 that referenced this pull request Jun 7, 2023
…-fit

add maxiter and maxfun optimisation arguments to LFR fit method
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants