zhiCHEN96 / ConceptWhitening

MIT License
117 stars 28 forks source link

Code for Fig. 3: Joint distribution of two axes #17

Closed houjunlin closed 1 year ago

houjunlin commented 1 year ago

Hi everyone! Thanks for sharing the code. I would like to visualize the joint distribution of two axes as shown in Fig. 3. Any code in the plot_functions.py file may I use to achieve this? Many thanks!

Best, JL

zhiCHEN96 commented 1 year ago

Hi Junlin,

You can use the plot_concept_representation function (line 892) in plot_functions.py

Best, Zhi

houjunlin commented 1 year ago

Hi Zhi, Thanks for your reply! I have run it successfully.