Learning transport cost from subset correspondence