Unsupervised sub-domain adaptation using optimal transport