def plot_mutual_info(m, labels):
    perm = min_weight_perm(-m)
    not_in_perm = [i for i in range(K) if perm not in i]
    permuted_m = np.hstack([m[:, perm], m[:, not_in_perm]])
    p = pcolormesh(permuted_m, xticklabels=perm + not_in_perm,
                   yticklabels=labels, cmap=CMAP)
    cb = plt.colorbar(p, drawedges=False)
    cb.outline.set_visible(False)


# Re-compute training coefficients for experiments
dataset/ETHPy150Open omangin/multimodal/samples/plot_info_matrix.py/plot_mutual_info/swapped operands of `not in`