Yes, theres a function called associations in the repo — the heat-maps you see here were plotted using it. There’s also an example of how to use it in the repo in example.py