search

alan-turing-institute / affinity-vae

Self-supervised method for disentanglement, clustering and classification of objects in multidimensional image data
BSD 3-Clause "New" or "Revised" License
13 stars 7 forks source link

Affinity matrix visualisation fails with matplotlib 3.9.0 #312

Open jolaem opened 5 months ago

jolaem commented 5 months ago

Failing with following error:

  File "/home/sxy26921/myruns/czi/01/../../../mygit/affinity-vae/run.py", line 693, in run
    run_pipeline(data)
  File "/home/sxy26921/myruns/czi/01/../../../mygit/affinity-vae/run.py", line 702, in run_pipeline
    train(
  File "/home/sxy26921/mygit/affinity-vae/avae/train.py", line 197, in train
    trains, vals, tests, affinity_matrix, data_dim = load_data(
  File "/home/sxy26921/mygit/affinity-vae/avae/data.py", line 169, in load_data
    affinity = get_affinity_matrix(affinity_path, classes_list)
  File "/home/sxy26921/mygit/affinity-vae/avae/data.py", line 282, in get_affinity_matrix
    plot_affinity_matrix(
  File "/home/sxy26921/mygit/affinity-vae/avae/vis.py", line 1890, in plot_affinity_matrix
    im = ax.imshow(lookup, vmin=-1, vmax=1, cmap=plt.cm.get_cmap("RdBu"))
AttributeError: module 'matplotlib.cm' has no attribute 'get_cmap'