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

Change the API for deep DA loss #171

Open tgnassou opened 2 months ago

tgnassou commented 2 months ago

In the DeepCoral method, the class DeepCoralLoss computes first the covariance matrices and then calls the deep_coral_loss function with cov as arguments. I would like to have to covariance estimation in the function deep_coral_loss to be able to use it alone.

Check if it is the same in the different deep DA methods.