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