-
Notifications
You must be signed in to change notification settings - Fork 21
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
[MRG] fix predict_proba
for deep method
#247
Conversation
Right now, it's not possible to do classification with only two classes in |
I added a test that checks when |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
looks a bit too magical for me...
predict_proba
for deep methodpredict_proba
for deep method
The
predict_proba
of deep method was choosing the nonlin attributes by looking atcriterion
attribute of the solvers. TheDomainAwareNet
has aDomainAwareCriterion
as criterion. We should choose the nonlin base oncriterion.base_criterion
. This PR does that.