Your approach is much more intuitive. I was coming back to say why didn't they show an embedding with categorical/scalar features?