Colour is the 3rd component -- I wanted to see the difference between overlapping episodes.

As for the 3D plot... here you go!

https://interconnected.org/more/2023/03/in_our_time-PCA-3D-p...

Basic PCA + Plotly is actually in OpenAI's official Python library (in `embedding_utils`) -- this plot is just the output from that.