Just in case if you want to do it in FastaiV2, you can check it out with this notebook:
I hope you will build your own heat map at the end.

Just in case if you want to do it in FastaiV2, you can check it out with this notebook:
I hope you will build your own heat map at the end.