Supervised Tree-Wasserstein Distance