Fix CDAN input to domain_classifier and gpu compatibility

scikit-adaptation / skada

Domain adaptation toolbox compatible with scikit-learn and pytorch

https://scikit-adaptation.github.io/

BSD 3-Clause "New" or "Revised" License

52 stars 16 forks source link

Fix CDAN input to domain_classifier and gpu compatibility #178

Closed Florent-Michel closed 2 months ago

Florent-Michel commented 2 months ago

The input size of domain_classifier is n_classes * n_features, so I propose to add n_classes as argument of CDAN to create domain_classifier with the right input size when it's not provided.