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 sample_weight parameter in TLCenter, TLStretch and TLRotate #273
Conversation
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 good to me !
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.
Thx @apmellot for this contribution!
Can you complete tests with sample weights different from None?
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 good but it needs tests
The tests now include sample_weight different from None. |
Ah! tests with weights failed https://github.com/pyRiemann/pyRiemann/actions/runs/7127343018/job/19407055709?pr=273#step:7:108 |
Hi there, thanks @apmellot for your help! I'm accepting this PR and would like to point out that |
@sylvchev @agramfort
Following the discussion we had. I added a sample_weight parameter in .fit and .fit_transform in the 3 transfer learning classes. I tested it, and it runs properly.