Add support for coloring points based on other features than class (e.g. cosine similarity and other embedding distances).